multi-agent-rlenv 3.3.2__tar.gz → 3.3.3__tar.gz

{multi_agent_rlenv-3.3.2 → multi_agent_rlenv-3.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: multi-agent-rlenv
-Version: 3.3.2
+Version: 3.3.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.3.2 → multi_agent_rlenv-3.3.3}/src/marlenv/__init__.py

@@ -62,7 +62,7 @@ print(env.extras_shape) # (1, )
 If you want to create a new environment, you can simply create a class that inherits from `MARLEnv`. If you want to create a wrapper around an existing `MARLEnv`, you probably want to subclass `RLEnvWrapper` which implements a default behaviour for every method.
 """
 
-__version__ = "3.3.2"
+__version__ = "3.3.3"
 
 from . import models
 from . import wrappers

{multi_agent_rlenv-3.3.2 → multi_agent_rlenv-3.3.3}/src/marlenv/adapters/overcooked_adapter.py

@@ -96,6 +96,10 @@ class Overcooked(MARLEnv[Sequence[int] | npt.NDArray, DiscreteActionSpace]):
         return {"horizon": self.horizon, "mdp": self._mdp}
 
     def __setstate__(self, state: dict):
+        from overcooked_ai_py.mdp.overcooked_mdp import Recipe
+
+        mdp = state["mdp"]
+        Recipe.configure(mdp.recipe_config)
         self.__init__(OvercookedEnv.from_mdp(state["mdp"], horizon=state["horizon"]))
 
     def get_image(self):

{multi_agent_rlenv-3.3.2 → multi_agent_rlenv-3.3.3}/tests/test_serialization.py

@@ -2,6 +2,7 @@ import pickle
 import numpy as np
 import orjson
 import pytest
+import os
 from copy import deepcopy
 
 import marlenv
@@ -156,3 +157,35 @@ def test_pickle_overcooked():
         step = env.step(actions)
         step_restored = restored.step(actions)
         assert step == step_restored
+
+
+@pytest.mark.skipif(not marlenv.adapters.HAS_OVERCOOKED, reason="Overcooked is not installed")
+def test_unpickling_from_blank_process():
+    from marlenv.adapters import Overcooked
+    import pickle
+    import subprocess
+    import tempfile
+
+    env = Overcooked.from_layout("large_room")
+    env_file = tempfile.NamedTemporaryFile("wb", delete=False)
+    pickle.dump(env, env_file)
+    env_file.close()
+
+    # Write the python file
+
+    f = tempfile.NamedTemporaryFile("w", delete=False)
+    f.write("""
+import pickle
+import sys
+
+with open(sys.argv[1], "rb") as f:
+    env = pickle.load(f)
+
+env.reset()""")
+    f.close()
+    try:
+        output = subprocess.run(f"python {f.name} {env_file.name}", shell=True, capture_output=True)
+        assert output.returncode == 0, output.stderr.decode("utf-8")
+    finally:
+        os.remove(f.name)
+        os.remove(env_file.name)