From 3e0b3e87edb6b69ec6a6a15cdfe64bb485fed31a Mon Sep 17 00:00:00 2001
From: nimishsantosh107 <nimishsantosh107@icloud.com>
Date: Wed, 15 Sep 2021 19:36:30 +0530
Subject: [PATCH] evaluator DONE status check updated - untested
---
flatland/evaluators/service.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/flatland/evaluators/service.py b/flatland/evaluators/service.py
index c896cde5..1ccbbd28 100644
--- a/flatland/evaluators/service.py
+++ b/flatland/evaluators/service.py
@@ -22,7 +22,7 @@ import timeout_decorator
import flatland
from flatland.core.env_observation_builder import DummyObservationBuilder
-from flatland.envs.agent_utils import RailAgentStatus
+from flatland.envs.step_utils.states import TrainState
from flatland.envs.malfunction_generators import malfunction_from_file
from flatland.envs.rail_env import RailEnv
from flatland.envs.rail_generators import rail_from_file
@@ -898,7 +898,7 @@ class FlatlandRemoteEvaluationService:
complete = 0
for i_agent in range(self.env.get_num_agents()):
agent = self.env.agents[i_agent]
- if agent.status in [RailAgentStatus.DONE_REMOVED]:
+ if agent.state == TrainState.DONE:
complete += 1
percentage_complete = complete * 1.0 / self.env.get_num_agents()
self.simulation_percentage_complete[-1] = percentage_complete
--
GitLab