ready for test

FLAIROx · Mar 15, 2024 · a535496 · a535496
1 parent aae9f99
commit a535496
Show file tree

Hide file tree

Showing 8 changed files with 1,211 additions and 910 deletions.
diff --git a/jaxmarl/environments/__init__.py b/jaxmarl/environments/__init__.py
@@ -17,8 +17,8 @@
 from .smax import SMAX, HeuristicEnemySMAX, LearnedPolicyEnemySMAX
 from .switch_riddle import SwitchRiddle
 from .overcooked import Overcooked, overcooked_layouts
-from .mabrax import Ant, Humanoid, Hopper, Walker2d, HalfCheetah
-from .hanabi import HanabiGame
+#from .mabrax import Ant, Humanoid, Hopper, Walker2d, HalfCheetah
+from .hanabi import Hanabi
 from .storm import InTheGrid, InTheGrid_2p
 from .coin_game import CoinGame
 
diff --git a/jaxmarl/environments/hanabi/__init__.py b/jaxmarl/environments/hanabi/__init__.py
@@ -1 +1 @@
-from .hanabi_obl import HanabiOBL as HanabiGame
+from .hanabi import HanabiEnv as Hanabi
diff --git a/jaxmarl/environments/hanabi/decks_test.json b/jaxmarl/environments/hanabi/decks_test.json