Add test dependency

LemonPi · LemonPi · commit f4db26aa5662 · 2024-02-14T15:20:12.000-05:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -54,6 +54,7 @@ dependencies = [
 test = [
     "pytest",
     "pybullet",
+    "jax",
 ]
 
 [project.urls]
