From a7e149a4c56b7c16b67c96511860ebf1f1838b52 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Thu, 3 Oct 2019 08:52:48 -0400
Subject: [PATCH] changed all node to city
---
flatland/envs/rail_generators.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index a25862dc..a1b8054a 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -1,7 +1,6 @@
"""Rail generators (infrastructure manager, "Infrastrukturbetreiber")."""
-import time
import warnings
-from typing import Callable, Tuple, Optional, Dict, List, Any
+from typing import Callable, Tuple, Optional, Dict, List
import msgpack
import numpy as np
@@ -592,7 +591,7 @@ def sparse_rail_generator(max_num_cities: int = 5, grid_mode: bool = False, max_
return grid_map, {'agents_hints': {
'num_agents': num_agents,
- 'agent_start_targets_nodes': agent_start_targets_cities,
+ 'agent_start_targets_cities': agent_start_targets_cities,
'train_stations': train_stations,
'city_orientations': city_orientations
}}
--
GitLab