fixed first tests in malfunction test

examples/flatland_2_0_example.py

@@ -37,7 +37,11 @@ env = RailEnv(width=100,
                                                    seed=14,  # Random seed
                                                    grid_mode=False,
                                                    max_rails_between_cities=2,
+<<<<<<< HEAD
                                                    max_rails_in_city=13,
+=======
+                                                   max_rails_in_city=8,
+>>>>>>> fixed first tests in malfunction test
                                                    ),
               schedule_generator=sparse_schedule_generator(speed_ration_map),
               number_of_agents=100,

flatland/envs/distance_map.py

@@ -55,6 +55,7 @@ class DistanceMap:
         self.env_width = rail.width
 
     def _compute(self, agents: List[EnvAgent], rail: GridTransitionMap):
+        print("computing distance map")
         self.agents_previous_computation = self.agents
         self.distance_map = np.inf * np.ones(shape=(len(agents),
                                                     self.env_height,

tests/test_flatland_malfunction.py

@@ -82,7 +82,11 @@ def test_malfunction_process():
                   obs_builder_object=SingleAgentNavigationObs()
                   )
     # reset to initialize agents_static
+<<<<<<< HEAD
     obs, info = env.reset(False, False, True, random_seed=10)
+=======
+    obs, info = env.reset(False, False, True, random_seed=0)
+>>>>>>> fixed first tests in malfunction test
     print(env.agents[0].malfunction_data)
     # Check that a initial duration for malfunction was assigned
     assert env.agents[0].malfunction_data['next_malfunction'] > 0
@@ -151,7 +155,11 @@ def test_malfunction_process_statistically():
                   obs_builder_object=SingleAgentNavigationObs()
                   )
     # reset to initialize agents_static
+<<<<<<< HEAD
     env.reset(True, True, False, random_seed=10)
+=======
+    env.reset(False, False, False, random_seed=0)
+>>>>>>> fixed first tests in malfunction test
 
     env.agents[0].target = (0, 0)
     nb_malfunction = 0