Skip to content
Snippets Groups Projects
Commit d8389906 authored by Erik Nygren's avatar Erik Nygren :bullettrain_front:
Browse files

Merge branch '288_preparing_submission_files' into 'master'

this closes #288

Closes #288

See merge request !266
parents 5bab56d3 fa6ac443
No related branches found
No related tags found
No related merge requests found
...@@ -11,6 +11,7 @@ import numpy as np ...@@ -11,6 +11,7 @@ import numpy as np
import redis import redis
import flatland import flatland
from envs.malfunction_generators import malfunction_from_file
from flatland.envs.rail_env import RailEnv from flatland.envs.rail_env import RailEnv
from flatland.envs.rail_generators import rail_from_file from flatland.envs.rail_generators import rail_from_file
from flatland.envs.schedule_generators import schedule_from_file from flatland.envs.schedule_generators import schedule_from_file
...@@ -217,6 +218,7 @@ class FlatlandRemoteClient(object): ...@@ -217,6 +218,7 @@ class FlatlandRemoteClient(object):
self.current_env_path = test_env_file_path self.current_env_path = test_env_file_path
self.env = RailEnv(width=1, height=1, rail_generator=rail_from_file(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), 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) obs_builder_object=obs_builder_object)
time_start = time.time() time_start = time.time()
......
...@@ -16,6 +16,7 @@ import redis ...@@ -16,6 +16,7 @@ import redis
import timeout_decorator import timeout_decorator
import flatland import flatland
from envs.malfunction_generators import malfunction_from_file
from flatland.core.env_observation_builder import DummyObservationBuilder from flatland.core.env_observation_builder import DummyObservationBuilder
from flatland.envs.agent_utils import RailAgentStatus from flatland.envs.agent_utils import RailAgentStatus
from flatland.envs.rail_env import RailEnv from flatland.envs.rail_env import RailEnv
...@@ -337,6 +338,7 @@ class FlatlandRemoteEvaluationService: ...@@ -337,6 +338,7 @@ class FlatlandRemoteEvaluationService:
del self.env del self.env
self.env = RailEnv(width=1, height=1, rail_generator=rail_from_file(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), 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()) obs_builder_object=DummyObservationBuilder())
if self.begin_simulation: if self.begin_simulation:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment