fixed future warning

flatironinstitute · May 17, 2024 · 136d21a · 136d21a
1 parent df1a7b0
commit 136d21a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/docs/api_guide/plot_05_batch_glm.py b/docs/api_guide/plot_05_batch_glm.py
@@ -99,7 +99,7 @@ def batcher():
 #
 # Let's do a few iterations of gradient descent calling the `batcher` function at every step.
 # At each step, we store the log-likelihood of the model for each neuron evaluated on the batch
-n_step = 500
+n_step = 5000
 logl = np.zeros(n_step)
 
 for i in range(n_step):	
@@ -163,7 +163,7 @@ def batcher():
 # %%
 # Let's compare scores for each neurons as well as the coefficients.
 
-plt.figure(figsize=(10, 10))
+plt.figure(figsize=(10, 8))
 gs = plt.GridSpec(3,2)
 plt.subplot(gs[0,:])
 plt.bar(np.arange(0, n_neurons), full_scores, 0.4, label="Full model")

diff --git a/pyproject.toml b/pyproject.toml
@@ -22,7 +22,7 @@ classifiers = [
 
 # Define dependencies for the project
 dependencies = [
-    "jax>=0.4",                     # Numerical computing library
+    "jax>=0.4.28",                     # Numerical computing library
     "jaxopt>=0.6",                  # Optimization library built on JAX
     "numpy>1.20",                   # Numerical computing library
     "scipy>=1.10",                  # Scientific computing library