updated file for conflict detection training. Used for introduction to the problem

torch_training/multi_agent_training.py

@@ -43,9 +43,11 @@ def main(argv):
     tree_depth = 3
     print("main2")
 
-    # Get an observation builder and predictor
-    # The predictor will always predict the shortest path from the current location of the agent.
-    # This is used to warn for potential conflicts --> Should be enhanced to get better performance!
+    """
+     Get an observation builder and predictor:
+     The predictor will always predict the shortest path from the current location of the agent.
+     This is used to warn for potential conflicts --> Should be enhanced to get better performance!
+    """
     predictor = ShortestPathPredictorForRailEnv()
     observation_helper = TreeObsForRailEnv(max_depth=tree_depth, predictor=predictor)