From 4d8c5cde55a4bde62655034f1eca80a9258f2665 Mon Sep 17 00:00:00 2001 From: Felipe Montealegre-Mora Date: Fri, 5 Jul 2024 18:26:46 +0000 Subject: [PATCH] np.inf discontinued (pt3) --- 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..789e36f 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, len(pop)), None) ) diff --git a/src/rl4caribou/envs/caribou_ode.py b/src/rl4caribou/envs/caribou_ode.py index 34cd5c5..09c4c94 100644 --- 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) )