From daca3092bd02ee6c919de34337aaf3f93f6bf90a Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Sat, 15 Jun 2019 13:34:52 +0200
Subject: [PATCH] fixex formatting
---
flatland/envs/observations.py | 3 ---
flatland/envs/predictions.py | 1 -
2 files changed, 4 deletions(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 007f6939..4de614db 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -578,11 +578,8 @@ class TreeObsForRailEnv(ObservationBuilder):
pow4 *= 4
child_size = (len(tree) - num_features_per_node) // 4
tree_data = tree[:4].tolist()
- # print("data",tree_data)
distance_data = [tree[4]]
- #print("distance",distance_data)
agent_data = tree[5:num_features_per_node].tolist()
- #print("agent_data",agent_data)
for children in range(4):
child_tree = tree[(num_features_per_node + children * child_size):
(num_features_per_node + (children + 1) * child_size)]
diff --git a/flatland/envs/predictions.py b/flatland/envs/predictions.py
index cff4e00e..b6fe8631 100644
--- a/flatland/envs/predictions.py
+++ b/flatland/envs/predictions.py
@@ -108,7 +108,6 @@ class ShortestPathPredictorForRailEnv(PredictionBuilder):
prediction_dict = {}
agent_idx = 0
for agent in agents:
- action_priorities = [RailEnvActions.MOVE_FORWARD, RailEnvActions.MOVE_LEFT, RailEnvActions.MOVE_RIGHT]
_agent_initial_position = agent.position
_agent_initial_direction = agent.direction
prediction = np.zeros(shape=(self.max_depth + 1, 5))
--
GitLab