evolutionary-policy-optimization 0.0.34__tar.gz → 0.0.35__tar.gz

{evolutionary_policy_optimization-0.0.34 → evolutionary_policy_optimization-0.0.35}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: evolutionary-policy-optimization
-Version: 0.0.34
+Version: 0.0.35
 Summary: EPO - Pytorch
 Project-URL: Homepage, https://pypi.org/project/evolutionary-policy-optimization/
 Project-URL: Repository, https://github.com/lucidrains/evolutionary-policy-optimization

{evolutionary_policy_optimization-0.0.34 → evolutionary_policy_optimization-0.0.35}/evolutionary_policy_optimization/epo.py

@@ -558,6 +558,7 @@ class LatentGenePool(Module):
         state: Tensor | None = None,
         latent_id: int | None = None,
         net: Module | None = None,
+        net_latent_kwarg_name = 'latent',
         **kwargs,
     ):
         device = self.latents.device
@@ -583,9 +584,11 @@ class LatentGenePool(Module):
         if not exists(net):
             return latent
 
+        latent_kwarg = {net_latent_kwarg_name: latent}
+
         return net(
             *args,
-            latent = latent,
+            **latent_kwarg,
             **kwargs
         )
 

{evolutionary_policy_optimization-0.0.34 → evolutionary_policy_optimization-0.0.35}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "evolutionary-policy-optimization"
-version = "0.0.34"
+version = "0.0.35"
 description = "EPO - Pytorch"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }