Commit daca3092 authored by Erik Nygren's avatar Erik Nygren
Browse files

fixex formatting

parent ff2ea11d
Pipeline #1105 failed with stages
in 9 minutes and 18 seconds
......@@ -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)]
......
......@@ -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))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment