#42 run baselines in ci

torch_training/bla.py

@@ -29,6 +29,32 @@ def main(argv):
         if opt in ('-n', '--n_trials'):
             n_trials = int(arg)
     print("main1")
+    random.seed(1)
+    np.random.seed(1)
+
+    """
+    env = RailEnv(width=10,
+                  height=20, obs_builder_object=TreeObsForRailEnv(max_depth=3, predictor=ShortestPathPredictorForRailEnv()))
+    env.load("./railway/complex_scene.pkl")
+    file_load = True
+    """
+
+    x_dim = np.random.randint(8, 20)
+    y_dim = np.random.randint(8, 20)
+    n_agents = np.random.randint(3, 8)
+    n_goals = n_agents + np.random.randint(0, 3)
+    min_dist = int(0.75 * min(x_dim, y_dim))
+    print("main2")
+
+    env = RailEnv(width=x_dim,
+                  height=y_dim,
+                  rail_generator=complex_rail_generator(nr_start_goal=n_goals, nr_extra=5, min_dist=min_dist,
+                                                        max_dist=99999,
+                                                        seed=0),
+                  obs_builder_object=TreeObsForRailEnv(max_depth=3, predictor=ShortestPathPredictorForRailEnv()),
+                  number_of_agents=n_agents)
+    env.reset(True, True)
+    file_load = False
 
 print("multi_agent_trainging.py (2)")