debug

boettiger-lab · Feb 28, 2024 · 157ec1b · 157ec1b
1 parent cd6af7c
commit 157ec1b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/rl4caribou/agents/const_escapement.py b/src/rl4caribou/agents/const_escapement.py
@@ -19,7 +19,7 @@ def __init__(self, escapement_vec, env = None):
     def predict(self, observation, **kwargs):
         obs_nat_units = self.bound * self.to_01(observation)
         m_mort = self.moose_mortality(obs_nat_units[0])
-        w_mort = self.wolf_mortality(obs_nat_units[3])
+        w_mort = self.wolf_mortality(obs_nat_units[2])
         mortality = np.float32([m_mort, w_mort])
         return self.to_pm1(mortality), {}