np.inf discontinued (pt3)

boettiger-lab · Jul 5, 2024 · 4d8c5cd · 4d8c5cd
1 parent b78766f
commit 4d8c5cd
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, len(pop)), np.repeat(None, len(pop))
+            pop, np.repeat(0, len(pop)), None)
         )
 
 

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