diff --git a/flatland/evaluators/client.py b/flatland/evaluators/client.py index 922f7fc03700b9f2c4b8065c5271d6814a67a2eb..2d4be5589c15f666b0341851276fd41b4336505c 100644 --- a/flatland/evaluators/client.py +++ b/flatland/evaluators/client.py @@ -11,6 +11,7 @@ import numpy as np import redis import flatland +from envs.malfunction_generators import malfunction_from_file from flatland.envs.rail_env import RailEnv from flatland.envs.rail_generators import rail_from_file from flatland.envs.schedule_generators import schedule_from_file @@ -217,6 +218,7 @@ class FlatlandRemoteClient(object): self.current_env_path = test_env_file_path self.env = RailEnv(width=1, height=1, rail_generator=rail_from_file(test_env_file_path), schedule_generator=schedule_from_file(test_env_file_path), + malfunction_generator_and_process_data=malfunction_from_file(test_env_file_path), obs_builder_object=obs_builder_object) time_start = time.time() diff --git a/flatland/evaluators/service.py b/flatland/evaluators/service.py index 8a70f1973d1e3974083be5c72e6a3263c0601992..5dd65935bea9edfeb88482b9627d3628479d7547 100644 --- a/flatland/evaluators/service.py +++ b/flatland/evaluators/service.py @@ -16,6 +16,7 @@ import redis import timeout_decorator import flatland +from envs.malfunction_generators import malfunction_from_file from flatland.core.env_observation_builder import DummyObservationBuilder from flatland.envs.agent_utils import RailAgentStatus from flatland.envs.rail_env import RailEnv @@ -337,6 +338,7 @@ class FlatlandRemoteEvaluationService: del self.env self.env = RailEnv(width=1, height=1, rail_generator=rail_from_file(test_env_file_path), schedule_generator=schedule_from_file(test_env_file_path), + malfunction_generator_and_process_data=malfunction_from_file(test_env_file_path), obs_builder_object=DummyObservationBuilder()) if self.begin_simulation: