From e4d68d6c294f4fb9f0ca9f2453fea645b2de0334 Mon Sep 17 00:00:00 2001
From: Felipe Montealegre-Mora <34276401+felimomo@users.noreply.github.com>
Date: Fri, 5 Jul 2024 11:24:12 -0700
Subject: [PATCH] Revert "np.inf discontinued"

---
 src/rl4caribou/envs/caribou.py     | 2 +-
 src/rl4caribou/envs/caribou_ode.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/rl4caribou/envs/caribou.py b/src/rl4caribou/envs/caribou.py
index e4068ed..92385ec 100644
--- 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, pop.__len__()), np.repeat(np.Inf, pop.__len__())
         )
 
 
diff --git a/src/rl4caribou/envs/caribou_ode.py b/src/rl4caribou/envs/caribou_ode.py
index b647481..6f2576a 100644
--- 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, len(pop)), np.repeat(None, len(pop))
+            pop, np.repeat(0, pop.__len__()), np.repeat(np.Inf, pop.__len__())
         )