evolutionary-policy-optimization 0.0.71__tar.gz → 0.0.72__tar.gz

{evolutionary_policy_optimization-0.0.71 → evolutionary_policy_optimization-0.0.72}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: evolutionary-policy-optimization
-Version: 0.0.71
+Version: 0.0.72
 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.71 → evolutionary_policy_optimization-0.0.72}/evolutionary_policy_optimization/epo.py

@@ -1281,9 +1281,14 @@ class EPO(Module):
         self,
         agent: Agent,
         env,
-        num_learning_cycles
+        num_learning_cycles,
+        seed = None
     ):
 
+        if exists(seed):
+            torch.manual_seed(seed)
+            np.random.seed(seed)
+
         for _ in tqdm(range(num_learning_cycles), desc = 'learning cycle'):
 
             memories = self.gather_experience_from(env)

{evolutionary_policy_optimization-0.0.71 → evolutionary_policy_optimization-0.0.72}/pyproject.toml

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