Post-rebase fix imports.

flatland-association · Feb 13, 2025 · 8883e1c · 8883e1c
1 parent 1929d82
commit 8883e1c
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/tests/test_flatland_malfunction.py b/tests/test_flatland_malfunction.py
@@ -15,11 +15,6 @@
 from flatland.utils.simple_rail import make_simple_rail2
 from tests.test_utils import Replay, ReplayConfig, run_replay_config, set_penalties_for_replay
 
-from flatland.envs.step_utils.speed_counter import SpeedCounter
-from flatland.envs.step_utils.states import TrainState
-from flatland.utils.simple_rail import make_simple_rail2
-from test_utils import Replay, ReplayConfig, run_replay_config, set_penalties_for_replay
-
 
 class SingleAgentNavigationObs(ObservationBuilder):
     """