multi-agent-rlenv 3.3.1__py3-none-any.whl → 3.3.2__py3-none-any.whl

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.1"
+__version__ = "3.3.2"
 
 from . import models
 from . import wrappers

marlenv/adapters/overcooked_adapter.py

@@ -1,6 +1,7 @@
 import sys
 from dataclasses import dataclass
 from typing import Literal, Sequence
+from copy import deepcopy
 
 import cv2
 import numpy as np
@@ -87,6 +88,16 @@ class Overcooked(MARLEnv[Sequence[int] | npt.NDArray, DiscreteActionSpace]):
             info=info,
         )
 
+    def __deepcopy__(self, memo: dict):
+        mdp = deepcopy(self._mdp)
+        return Overcooked(OvercookedEnv.from_mdp(mdp, horizon=self.horizon))
+
+    def __getstate__(self):
+        return {"horizon": self.horizon, "mdp": self._mdp}
+
+    def __setstate__(self, state: dict):
+        self.__init__(OvercookedEnv.from_mdp(state["mdp"], horizon=state["horizon"]))
+
     def get_image(self):
         rewards_dict = {}  # dictionary of details you want rendered in the UI
         for key, value in self._oenv.game_stats.items():

{multi_agent_rlenv-3.3.1.dist-info → multi_agent_rlenv-3.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: multi-agent-rlenv
-Version: 3.3.1
+Version: 3.3.2
 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.1.dist-info → multi_agent_rlenv-3.3.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-marlenv/__init__.py,sha256=7HYiLVpZ4PQQWYjvc_78eXrnidv-FfUiY1pUkbxpm5U,3741
+marlenv/__init__.py,sha256=GFEcoE8jkA0vbuQyrVRpgsS-iLIelxjpwsB_6pGJGjs,3741
 marlenv/env_builder.py,sha256=_rdwcWRqnHP7i4M4Oje1Y2nrEBKH9EzTpqOuw_PNUyw,5560
 marlenv/env_pool.py,sha256=R3WIrnQ5Zvff4HR1ecfkDmuO2zl7v1ywQ0K2_nvWFzs,1070
 marlenv/exceptions.py,sha256=gJUC_2rVAvOfK_ypVFc7Myh-pIfSU3To38VBVS_0rZA,1179
@@ -6,7 +6,7 @@ marlenv/mock_env.py,sha256=qB0fYFIfbopJf7Va8kCeVI5vsOy1-2JdEYe9gdV1Ruw,4761
 marlenv/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 marlenv/adapters/__init__.py,sha256=NEmuHPWz4SGQcgF7QuIeA0QaXK141JoYco-7mqj9Ghk,883
 marlenv/adapters/gym_adapter.py,sha256=Vx6ZrYI7kiNlJODmqyjXu9WCdbCr6trcMNot0pvYD74,2864
-marlenv/adapters/overcooked_adapter.py,sha256=exNYFQhtnrvXQtG7M7FuQUkcRJK60a8vu2uwhRLc0K4,5768
+marlenv/adapters/overcooked_adapter.py,sha256=Yf1xmjSgDOOSzR3QNqjxQ_ROFEdKUe_C8BF2nLlW3Us,6162
 marlenv/adapters/pettingzoo_adapter.py,sha256=9rwSc_b7qV3ChtEIevOkJvtIp7WoY3CVnu6L9DxlMB4,2852
 marlenv/adapters/pymarl_adapter.py,sha256=x__E90XpFbfSWhnBHtkcD6WYkmKki1LByNbUFoDBUcg,3416
 marlenv/adapters/smac_adapter.py,sha256=fOfKo1hL4ioKtM5qQGcwtfdkdwUEACjAZqaGmkoQUcU,8373
@@ -31,7 +31,7 @@ marlenv/wrappers/penalty_wrapper.py,sha256=v4_H8OEN2-yujLzRb6P7W7KwmXHtjAFsxcdp3
 marlenv/wrappers/rlenv_wrapper.py,sha256=C2XekgBIM4x3Wa2Mtsn7rihRD4ymC2hORI473Af0sfw,2962
 marlenv/wrappers/time_limit.py,sha256=CDIMMJPMyIDHSFxUJaC7nb7Kd86-07NgZeFhrpZm82o,3985
 marlenv/wrappers/video_recorder.py,sha256=d5AFu6qHqby9mOcBsYWYPxAPiK1vtnfMYdZ81AnCekI,2624
-multi_agent_rlenv-3.3.1.dist-info/METADATA,sha256=lrE00NibOxZAoMD_DacNbfIbpEHqX7zw5wXc0xj8iiY,4897
-multi_agent_rlenv-3.3.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-multi_agent_rlenv-3.3.1.dist-info/licenses/LICENSE,sha256=_eeiGVoIJ7kYt6l1zbIvSBQppTnw0mjnYk1lQ4FxEjE,1074
-multi_agent_rlenv-3.3.1.dist-info/RECORD,,
+multi_agent_rlenv-3.3.2.dist-info/METADATA,sha256=s9O2h4QdJZ4Ytq1hRiBFJPsAtWV0bd1JLxDx4MChaI0,4897
+multi_agent_rlenv-3.3.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+multi_agent_rlenv-3.3.2.dist-info/licenses/LICENSE,sha256=_eeiGVoIJ7kYt6l1zbIvSBQppTnw0mjnYk1lQ4FxEjE,1074
+multi_agent_rlenv-3.3.2.dist-info/RECORD,,