bug fix saving prey list correctly

artificial-life-lab · May 14, 2024 · 54c1ba2 · 54c1ba2
1 parent cd17dfd
commit 54c1ba2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/causal_inference/base/lotka_volterra.py b/causal_inference/base/lotka_volterra.py
@@ -65,7 +65,7 @@ def runge_kutta_update(self, current_prey, current_predators):
     def _save_population(self):
         filename = os.path.join(RESULTS_DIR, 'populations.h5')
         hf = h5py.File(filename, 'w')
-        hf.create_dataset('prey_pop', data=self.predator_list)
+        hf.create_dataset('prey_pop', data=self.prey_list)
         hf.create_dataset('pred_pop', data=self.predator_list)
         hf.close()