diff --git a/src/rl4caribou/envs/caribou.py b/src/rl4caribou/envs/caribou.py index 789e36f..d951405 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)), None) + pop, np.repeat(0, len(pop)), None ) diff --git a/src/rl4caribou/envs/caribou_ode.py b/src/rl4caribou/envs/caribou_ode.py index 09c4c94..878dc25 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)), None) + pop, np.repeat(0, len(pop)), None )