evolutionary-policy-optimization 0.0.17__tar.gz → 0.0.18__tar.gz

{evolutionary_policy_optimization-0.0.17 → evolutionary_policy_optimization-0.0.18}/.github/workflows/test.yml

@@ -8,8 +8,10 @@ jobs:
       - uses: actions/checkout@v4
       - name: Install Python
         uses: actions/setup-python@v4
+      - name: Install uv
+        run: pip install uv && uv venv
       - name: Install dependencies
         run: |
-          pip install .[test]
+          uv pip install .[test]
       - name: Run pytest
-        run: pytest tests
+        run: uv run pytest tests

{evolutionary_policy_optimization-0.0.17 → evolutionary_policy_optimization-0.0.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: evolutionary-policy-optimization
-Version: 0.0.17
+Version: 0.0.18
 Summary: EPO - Pytorch
 Project-URL: Homepage, https://pypi.org/project/evolutionary-policy-optimization/
 Project-URL: Repository, https://github.com/lucidrains/evolutionary-policy-optimization
@@ -33,7 +33,7 @@ Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Requires-Python: >=3.8
+Requires-Python: >=3.9
 Requires-Dist: adam-atan2-pytorch
 Requires-Dist: assoc-scan
 Requires-Dist: einops>=0.8.0
@@ -44,10 +44,6 @@ Provides-Extra: examples
 Requires-Dist: numpy; extra == 'examples'
 Requires-Dist: pufferlib>=2.0.6; extra == 'examples'
 Requires-Dist: tqdm; extra == 'examples'
-Provides-Extra: examples-gym
-Requires-Dist: box2d-py; extra == 'examples-gym'
-Requires-Dist: gymnasium[box2d]>=1.0.0; extra == 'examples-gym'
-Requires-Dist: tqdm; extra == 'examples-gym'
 Provides-Extra: test
 Requires-Dist: pytest; extra == 'test'
 Description-Content-Type: text/markdown

{evolutionary_policy_optimization-0.0.17 → evolutionary_policy_optimization-0.0.18}/evolutionary_policy_optimization/epo.py

@@ -206,6 +206,8 @@ class Actor(Module):
         assert len(dim_hiddens) >= 2
         dim_first, *_, dim_last = dim_hiddens
 
+        self.dim_latent = dim_latent
+
         self.init_layer = nn.Sequential(
             nn.Linear(dim_state, dim_first),
             nn.SiLU()
@@ -242,6 +244,8 @@ class Critic(Module):
         assert len(dim_hiddens) >= 2
         dim_first, *_, dim_last = dim_hiddens
 
+        self.dim_latent = dim_latent
+
         self.init_layer = nn.Sequential(
             nn.Linear(dim_state, dim_first),
             nn.SiLU()
@@ -490,6 +494,8 @@ class Agent(Module):
 
         self.latent_gene_pool = latent_gene_pool
 
+        assert actor.dim_latent == critic.dim_latent == latent_gene_pool.dim_latent
+
         # optimizers
 
         self.actor_optim = optim_klass(actor.parameters(), lr = actor_lr, **actor_optim_kwargs)

{evolutionary_policy_optimization-0.0.17 → evolutionary_policy_optimization-0.0.18}/pyproject.toml

@@ -1,12 +1,12 @@
 [project]
 name = "evolutionary-policy-optimization"
-version = "0.0.17"
+version = "0.0.18"
 description = "EPO - Pytorch"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }
 ]
 readme = "README.md"
-requires-python = ">= 3.8"
+requires-python = ">= 3.9"
 license = { file = "LICENSE" }
 keywords = [
     'artificial intelligence',
@@ -43,12 +43,6 @@ examples = [
     "tqdm",
 ]
 
-examples_gym = [
-    "box2d-py",
-    "gymnasium[box2d]>=1.0.0",
-    "tqdm",
-]
-
 test = [
     "pytest"
 ]

evolutionary_policy_optimization-0.0.18/requirements.txt

@@ -0,0 +1,3 @@
+box2d-py
+gymnasium[box2d]>=1.0.0
+tqdm