evolutionary-policy-optimization 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

evolutionary_policy_optimization/__init__.py

@@ -9,3 +9,7 @@ from evolutionary_policy_optimization.epo import (
 )
 
 from evolutionary_policy_optimization.mock_env import Env
+
+from evolutionary_policy_optimization.env_wrappers import (
+    GymnasiumEnvWrapper
+)

evolutionary_policy_optimization/env_wrappers.py

@@ -0,0 +1,36 @@
+import torch
+from torch.nn import Module
+
+from evolutionary_policy_optimization.epo import create_agent, Agent
+
+class GymnasiumEnvWrapper(Module):
+    def __init__(
+        self,
+        env
+    ):
+        super().__init__()
+        self.env = env
+
+    def reset(self, *args, **kwargs):
+        return self.env.reset(*args, **kwargs)
+
+    def step(self, *args, **kwargs):
+        return self.env.step(*args, **kwargs)
+
+    def to_agent_hparams(self):
+        return dict(
+            dim_state = self.env.observation_space.shape[0],
+            actor_num_actions = self.env.action_space.n
+        )
+
+    def to_epo_agent(
+        self,
+        *args,
+        **kwargs
+    ) -> Agent:
+
+        return create_agent(
+            *args,
+            **self.to_agent_hparams(),
+            **kwargs
+        )

evolutionary_policy_optimization/epo.py

@@ -1040,6 +1040,7 @@ def actor_loss(
 # agent contains the actor, critic, and the latent genetic pool
 
 def create_agent(
+    *,
     dim_state,
     num_latents,
     dim_latent,

{evolutionary_policy_optimization-0.1.0.dist-info → evolutionary_policy_optimization-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: evolutionary-policy-optimization
-Version: 0.1.0
+Version: 0.1.1
 Summary: EPO - Pytorch
 Project-URL: Homepage, https://pypi.org/project/evolutionary-policy-optimization/
 Project-URL: Repository, https://github.com/lucidrains/evolutionary-policy-optimization
@@ -34,6 +34,7 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.9
+Requires-Dist: accelerate
 Requires-Dist: adam-atan2-pytorch
 Requires-Dist: assoc-scan>=0.0.2
 Requires-Dist: einops>=0.8.1

evolutionary_policy_optimization-0.1.1.dist-info/RECORD

@@ -0,0 +1,10 @@
+evolutionary_policy_optimization/__init__.py,sha256=NyiYDYU7DlpmOTM7xiBQET3r1WwX0ebrgMCBLSQrW3c,288
+evolutionary_policy_optimization/distributed.py,sha256=7KgZdeS_wxBHo_du9XZFB1Cu318J-Bp66Xdr6Log_20,2423
+evolutionary_policy_optimization/env_wrappers.py,sha256=bDL06o9_b1iW6k3fw2xifnOnYlzs643tdW6Yv2gsIdw,803
+evolutionary_policy_optimization/epo.py,sha256=hxXOUt4tCmlqN2zjvG6Qfp_icq2ApuMowx3xK0Wm3wY,39498
+evolutionary_policy_optimization/experimental.py,sha256=-IgqjJ_Wk_CMB1y9YYWpoYqTG9GZHAS6kbRdTluVevg,1563
+evolutionary_policy_optimization/mock_env.py,sha256=TLyyRm6tOD0Kdn9QqJJQriaSnsR-YmNQHo4OohmZFG4,1410
+evolutionary_policy_optimization-0.1.1.dist-info/METADATA,sha256=ToFlj-YAqNqTZx2W9LZXmhteQBBUCTO00uAFd-p9s48,6329
+evolutionary_policy_optimization-0.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+evolutionary_policy_optimization-0.1.1.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
+evolutionary_policy_optimization-0.1.1.dist-info/RECORD,,

evolutionary_policy_optimization-0.1.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-evolutionary_policy_optimization/__init__.py,sha256=0q0aBuFgWi06MLMD8FiHzBYQ3_W4LYWrwmCtF3u5H2A,201
-evolutionary_policy_optimization/distributed.py,sha256=7KgZdeS_wxBHo_du9XZFB1Cu318J-Bp66Xdr6Log_20,2423
-evolutionary_policy_optimization/epo.py,sha256=VrFD5lFQrS7KeYTC-WavEMTHgQXoq7vNPVXnRJwFSDI,39491
-evolutionary_policy_optimization/experimental.py,sha256=-IgqjJ_Wk_CMB1y9YYWpoYqTG9GZHAS6kbRdTluVevg,1563
-evolutionary_policy_optimization/mock_env.py,sha256=TLyyRm6tOD0Kdn9QqJJQriaSnsR-YmNQHo4OohmZFG4,1410
-evolutionary_policy_optimization-0.1.0.dist-info/METADATA,sha256=Nl3JSTSireDXXvX1ZrMQqTA8CXz1USUQOwPuuCtgCJw,6303
-evolutionary_policy_optimization-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-evolutionary_policy_optimization-0.1.0.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
-evolutionary_policy_optimization-0.1.0.dist-info/RECORD,,