From 7214368ce5752902609de70eb798a9cf6354db37 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Mon, 19 Aug 2019 22:34:05 +0200
Subject: [PATCH] formating
---
flatland/envs/generators.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index 80c9f13b..fc5f9fab 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -790,10 +790,11 @@ def realistic_rail_generator(nr_start_goal=1, seed=0):
if len(new_path) > 0:
c = (pos_x - 1, pos_y - 1)
make_switch_e_w(width, height, grid_map, c)
+ add_pos = (int((goal_track[0] + start_track[0]) / 2), int((goal_track[1] + start_track[1]) / 2), idx_target)
agents_positions_backward.append(add_pos)
- agents_directions_backward.append(1)
+ agents_directions_backward.append(3)
idx_backward.append(idx_target)
- add_pos = (goal_track[0], goal_track[1], idx_target)
+ add_pos = (int((goal_track[0] + start_track[0]) / 2), int((goal_track[1] + start_track[1]) / 2), idx_target)
agents_targets.append(add_pos)
idx_target += 1
else:
@@ -803,10 +804,11 @@ def realistic_rail_generator(nr_start_goal=1, seed=0):
if len(new_path) > 0:
c = (pos_x - 1, pos_y + 1)
make_switch_w_e(width, height, grid_map, c)
+ add_pos = (int((goal_track[0] + start_track[0]) / 2), int((goal_track[1] + start_track[1]) / 2), idx_target)
agents_positions_forward.append(add_pos)
- agents_directions_forward.append(3)
+ agents_directions_forward.append(1)
idx_forward.append(idx_target)
- add_pos = (goal_track[0], goal_track[1], idx_target)
+ add_pos = (int((goal_track[0] + start_track[0]) / 2), int((goal_track[1] + start_track[1]) / 2), idx_target)
agents_targets.append(add_pos)
idx_target += 1
--
GitLab