Fix a test

oist · Jun 6, 2024 · 605e1f5 · 605e1f5
1 parent 94d8ee5
commit 605e1f5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -24,6 +24,7 @@ dependencies = [
     "equinox >= 0.11",
     "moderngl >= 5.6",
     "moderngl-window >= 2.4",
+    # TODO: Update jax to >= 0.4.27 because clip has a breaking change
     "jax >= 0.4",
     "pyarrow >= 9.0",
     "pyserde[toml] >= 0.14",

diff --git a/tests/test_observe.py b/tests/test_observe.py
@@ -48,7 +48,7 @@ def reset_env(key: chex.PRNGKey) -> tuple[CircleForaging, CFState, TimeStep[CFOb
             [80.0, 90.0],
             [80.0, 120.0],
         ),
-        n_max_food_regen=10,
+        n_max_food_regen=20,
         food_num_fn=("constant", 3),
         foodloc_interval=20,
         agent_radius=AGENT_RADIUS,