#2648 - changed av_reward_per_episode to total_reward_per_episode in primaite_benchmark.py and report.py

2024-06-26 13:26:18 +01:00
parent e7f979b78e
commit 7f0a0c562f
2 changed files with 10 additions and 10 deletions
--- a/benchmark/primaite_benchmark.py
+++ b/benchmark/primaite_benchmark.py
@@ -124,7 +124,7 @@ class BenchmarkSession:
            "total_s": total_s,
            "s_per_step": s_per_step,
            "s_per_100_steps_10_nodes": s_per_100_steps_10_nodes,
-            "av_reward_per_episode": self.gym_env.total_reward_per_episode,
+            "total_reward_per_episode": self.gym_env.total_reward_per_episode,
        }