From d1708a319955aa9948504e0a76104934661bc4a7 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Mon, 15 Jul 2019 11:53:10 -0400
Subject: [PATCH] updated file for conflict detection training. Used for
introduction to the problem
---
torch_training/multi_agent_training.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/torch_training/multi_agent_training.py b/torch_training/multi_agent_training.py
index bee0d96..4d337d1 100644
--- a/torch_training/multi_agent_training.py
+++ b/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)
--
GitLab