multi-agent-rlenv 3.6.1__py3-none-any.whl → 3.6.3__py3-none-any.whl

marlenv/catalog/__init__.py

@@ -1,26 +1,22 @@
-from importlib.util import find_spec
-from ..utils.import_placeholders import dummy_type
 from marlenv.adapters import SMAC
 from .deepsea import DeepSea
 
 
-HAS_LLE = find_spec("lle") is not None
-if HAS_LLE:
-    from lle import LLE  # pyright: ignore[reportMissingImports]
-else:
-    LLE = dummy_type("lle", "laser-learning-environment")
-
-HAS_OVERCOOKED = find_spec("overcooked") is not None
-if HAS_OVERCOOKED:
-    from overcooked import Overcooked  # pyright: ignore[reportMissingImports]
-else:
-    Overcooked = dummy_type("overcooked", "overcooked")
-
 __all__ = [
-    "Overcooked",
     "SMAC",
-    "LLE",
     "DeepSea",
-    "HAS_LLE",
-    "HAS_OVERCOOKED",
+    "lle",
+    "overcooked",
 ]
+
+
+def lle():
+    from lle import LLE  # pyright: ignore[reportMissingImports]
+
+    return LLE
+
+
+def overcooked():
+    from overcooked import Overcooked  # pyright: ignore[reportMissingImports]
+
+    return Overcooked

marlenv/catalog/deepsea.py

@@ -32,9 +32,10 @@ class DeepSea(MARLEnv[MultiDiscreteSpace]):
         self._row = 0
         self._col = 0
         self._step_right_penalty = -0.01 / self.max_depth
+        self.name = f"{self.__class__.__name__}-{self.max_depth}"
 
     def get_observation(self) -> Observation:
-        return Observation(np.array([self._row, self._col], dtype=np.float32), self.available_actions())
+        return Observation(np.array([[self._row, self._col]], dtype=np.float32), self.available_actions())
 
     def get_state(self) -> State:
         return State(np.array([self._row, self._col], dtype=np.float32))

{multi_agent_rlenv-3.6.1.dist-info → multi_agent_rlenv-3.6.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: multi-agent-rlenv
-Version: 3.6.1
+Version: 3.6.3
 Summary: A strongly typed Multi-Agent Reinforcement Learning framework
 Project-URL: repository, https://github.com/yamoling/multi-agent-rlenv
 Author-email: Yannick Molinghen <yannick.molinghen@ulb.be>

{multi_agent_rlenv-3.6.1.dist-info → multi_agent_rlenv-3.6.3.dist-info}/RECORD

@@ -9,8 +9,8 @@ marlenv/adapters/gym_adapter.py,sha256=5HZF3g0QD4n7K4GQoMis4q0zj97uFTLdzdxMYHzM_
 marlenv/adapters/pettingzoo_adapter.py,sha256=w9Ta-X4L_6ZXdDGmREOdcU0vpLR8lGP__s49DyK3dk8,2852
 marlenv/adapters/pymarl_adapter.py,sha256=2s7EY31s1hrml3q-BBaXo_eDMXTjkebozZPvzsgrb9c,3353
 marlenv/adapters/smac_adapter.py,sha256=8uWC7YKsaSXeTS8AUhpGOKvrWMbVEQT2-pml5BaFUB0,8343
-marlenv/catalog/__init__.py,sha256=VMAlot7aHjCXJPH6iycDHuYIEUyK_giOIO8qWTN8H_Q,655
-marlenv/catalog/deepsea.py,sha256=VPRV6tjPIkj-TMqp2L8U9COUK3FBlodoRW6gpsnwZ9Y,2332
+marlenv/catalog/__init__.py,sha256=dlT_xXdjV9gdXc518xRXC4Np1TSpAMV5Wdwk4M2br8c,351
+marlenv/catalog/deepsea.py,sha256=kwmTkp_iUf1ltpYIcZoZVJKriWSAQa8z6pv7qy4jFDE,2400
 marlenv/models/__init__.py,sha256=uihmRs71Gg5z7Bvau_xtaQVg7xEtX8sTzi74bIHL5P0,443
 marlenv/models/env.py,sha256=BG1iVHxGD_p827mF0ewyOBn6wU2gtFsHLW1b4UtW-V0,7841
 marlenv/models/episode.py,sha256=zsyxsW4LIioPKyY4DZKn64A31e5ZvlwOf3HIGuRUzhs,13531
@@ -38,7 +38,7 @@ marlenv/wrappers/potential_shaping.py,sha256=T_QvnmWReCgpyoInxRw2UXbmdvcBD5U-vV1
 marlenv/wrappers/rlenv_wrapper.py,sha256=S6G1VjFklTEzU6bj0AXrTDXnsTQJARq8VB4uUH6AXe4,2993
 marlenv/wrappers/time_limit.py,sha256=GxbxcbfFyuVg14ylQU2C_cjmV9q4uDAt5wepfgX_PyM,3976
 marlenv/wrappers/video_recorder.py,sha256=ucBQSNRPqDr-2mYxrTCqlrWcxSWtSJ7XlRC9-LdukBM,2535
-multi_agent_rlenv-3.6.1.dist-info/METADATA,sha256=FS_Hq7KmVDmZTHze01jijOSlEUf_QTA0CsZMcO1pzkY,5751
-multi_agent_rlenv-3.6.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-multi_agent_rlenv-3.6.1.dist-info/licenses/LICENSE,sha256=_eeiGVoIJ7kYt6l1zbIvSBQppTnw0mjnYk1lQ4FxEjE,1074
-multi_agent_rlenv-3.6.1.dist-info/RECORD,,
+multi_agent_rlenv-3.6.3.dist-info/METADATA,sha256=jfW3dd1O5u8VurE32m6YTKeDXHU6iIDxF04G8sSisWM,5751
+multi_agent_rlenv-3.6.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+multi_agent_rlenv-3.6.3.dist-info/licenses/LICENSE,sha256=_eeiGVoIJ7kYt6l1zbIvSBQppTnw0mjnYk1lQ4FxEjE,1074
+multi_agent_rlenv-3.6.3.dist-info/RECORD,,