From e1ffa1b48872b41673555f61c4a7b01dcc27411d Mon Sep 17 00:00:00 2001
From: nimishsantosh107 <nimishsantosh107@icloud.com>
Date: Fri, 3 Sep 2021 19:22:39 +0530
Subject: [PATCH] changed default values for sparse_rail_generator
---
flatland/envs/rail_generators.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index 08d2df07..90dcfb36 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -162,7 +162,7 @@ def sparse_rail_generator(*args, **kwargs):
class SparseRailGen(RailGen):
- def __init__(self, max_num_cities: int = 5, grid_mode: bool = False, max_rails_between_cities: int = 4,
+ def __init__(self, max_num_cities: int = 2, grid_mode: bool = False, max_rails_between_cities: int = 2,
max_rail_pairs_in_city: int = 2, seed=0) -> RailGenerator:
"""
Generates railway networks with cities and inner city rails
--
GitLab