From 06e4795ddef1d19e64700c098892438a0f0b2449 Mon Sep 17 00:00:00 2001
From: Luca Grillotti <lg4615@ic.ac.uk>
Date: Mon, 9 Sep 2024 15:56:41 +0100
Subject: [PATCH] add import jumanji

---
 examples/jumanji_snake.ipynb | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/examples/jumanji_snake.ipynb b/examples/jumanji_snake.ipynb
index 21a236fa..78ec01c8 100644
--- a/examples/jumanji_snake.ipynb
+++ b/examples/jumanji_snake.ipynb
@@ -46,6 +46,8 @@
     "\n",
     "import functools\n",
     "\n",
+    "import jumanji\n",
+    "\n",
     "import numpy as np\n",
     "\n",
     "from qdax.baselines.genetic_algorithm import GeneticAlgorithm\n",