From 778c2585fbe4a289d4b9d47d816c42c5d5d56812 Mon Sep 17 00:00:00 2001 From: MLErik <baerenjesus@gmail.com> Date: Tue, 26 Nov 2019 16:33:02 -0500 Subject: [PATCH] renaming --- ...nerators.py => test_flatland_schedule_from_file.py} | 0 tests/test_utils.py | 10 +++------- 2 files changed, 3 insertions(+), 7 deletions(-) rename tests/{test_flatland_schedule_generators.py => test_flatland_schedule_from_file.py} (100%) diff --git a/tests/test_flatland_schedule_generators.py b/tests/test_flatland_schedule_from_file.py similarity index 100% rename from tests/test_flatland_schedule_generators.py rename to tests/test_flatland_schedule_from_file.py diff --git a/tests/test_utils.py b/tests/test_utils.py index bb344962..cae9b7fb 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -3,15 +3,13 @@ from typing import List, Tuple, Optional import numpy as np from attr import attrs, attrib -from flatland.envs.malfunction_generators import MalfunctionParameters, malfunction_from_params - -from flatland.envs.rail_generators import RailGenerator - -from flatland.envs.schedule_generators import ScheduleGenerator from flatland.core.grid.grid4 import Grid4TransitionsEnum from flatland.envs.agent_utils import EnvAgent, RailAgentStatus +from flatland.envs.malfunction_generators import MalfunctionParameters, malfunction_from_params from flatland.envs.rail_env import RailEnvActions, RailEnv +from flatland.envs.rail_generators import RailGenerator +from flatland.envs.schedule_generators import ScheduleGenerator from flatland.utils.rendertools import RenderTool @@ -139,8 +137,6 @@ def run_replay_config(env: RailEnv, test_configs: List[ReplayConfig], rendering: def create_and_save_env(file_name: str, schedule_generator: ScheduleGenerator, rail_generator: RailGenerator): - - stochastic_data = MalfunctionParameters(malfunction_rate=1000, # Rate of malfunction occurence min_duration=15, # Minimal duration of malfunction max_duration=50 # Max duration of malfunction -- GitLab