update version of Jumanji Snake environment

adaptive-intelligent-robotics · Nov 30, 2023 · 71e65e9 · 71e65e9
1 parent 965573f
commit 71e65e9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/jumanji_snake.ipynb b/examples/jumanji_snake.ipynb
@@ -91,7 +91,7 @@
    "outputs": [],
    "source": [
     "# Instantiate a Jumanji environment using the registry\n",
-    "env = jumanji.make('Snake-6x6-v0')\n",
+    "env = jumanji.make('Snake-v1')\n",
     "\n",
     "# Reset your (jit-able) environment\n",
     "key = jax.random.PRNGKey(0)\n",

diff --git a/tests/default_tasks_test/jumanji_envs_test.py b/tests/default_tasks_test/jumanji_envs_test.py
@@ -26,7 +26,7 @@ def test_jumanji_utils() -> None:
     batch_size = population_size
 
     # Instantiate a Jumanji environment using the registry
-    env = jumanji.make("Snake-6x6-v0")
+    env = jumanji.make("Snake-v1")
 
     # Reset your (jit-able) environment
     key = jax.random.PRNGKey(0)