diff --git a/flatland/evaluators/service.py b/flatland/evaluators/service.py index 60d8d461784b91cd66044e3a0d6835eafe07dd61..f4d92468f3eb89cf27db1136a1ddf56e73440fe1 100644 --- a/flatland/evaluators/service.py +++ b/flatland/evaluators/service.py @@ -614,6 +614,8 @@ class FlatlandRemoteEvaluationService: time_diff = time.time() - time_start self.update_running_stats("internal_env_step_time", time_diff) + self.current_step += 1 + cumulative_reward = sum(all_rewards.values()) self.simulation_rewards[-1] += cumulative_reward self.simulation_steps[-1] += 1