Merge pull request #114 from ali-shihab/main

Generalised STORM environment implementation.
FLAIROx · Sep 9, 2024 · 16a8f27 · 16a8f27
2 parents 699e9a0 + db32fe6
commit 16a8f27
Show file tree

Hide file tree

Showing 6 changed files with 2,281 additions and 2 deletions.
diff --git a/docs/imgs/storm_nplayer.gif b/docs/imgs/storm_nplayer.gif
diff --git a/jaxmarl/environments/__init__.py b/jaxmarl/environments/__init__.py
@@ -19,7 +19,7 @@
 from .overcooked import Overcooked, overcooked_layouts
 from .mabrax import Ant, Humanoid, Hopper, Walker2d, HalfCheetah
 from .hanabi import Hanabi
-from .storm import InTheGrid, InTheGrid_2p
+from .storm import InTheGrid, InTheGrid_2p, InTheMatrix
 from .coin_game import CoinGame
 from .jaxnav import JaxNav
 
diff --git a/jaxmarl/environments/storm/__init__.py b/jaxmarl/environments/storm/__init__.py
@@ -1,2 +1,3 @@
 from .storm_env import InTheGrid
-from .storm_2p import InTheGrid_2p
+from .storm_2p import InTheGrid_2p
+from .storm import InTheMatrix