Merge pull request #9 from boettiger-lab/nice-little-things

np.inf discontinued
boettiger-lab · Jul 5, 2024 · d8be324 · d8be324
2 parents 124f7e2 + b78766f
commit d8be324
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/rl4caribou/envs/caribou.py b/src/rl4caribou/envs/caribou.py
@@ -191,7 +191,7 @@ def state_units(self, pop):
     def population_units(self):
         pop = (self.state + 1) * self.bound / 2
         return np.clip(
-            pop, np.repeat(0, pop.__len__()), np.repeat(np.Inf, pop.__len__())
+            pop, np.repeat(0, len(pop)), np.repeat(None, len(pop))
         )
 
 

diff --git a/src/rl4caribou/envs/caribou_ode.py b/src/rl4caribou/envs/caribou_ode.py
@@ -274,7 +274,7 @@ def state_units(self, pop):
     def population_units(self):
         pop = (self.state + 1) * self.bound / 2
         return np.clip(
-            pop, np.repeat(0, pop.__len__()), np.repeat(np.Inf, pop.__len__())
+            pop, np.repeat(0, len(pop)), np.repeat(None, len(pop))
         )