diff --git a/flatland/envs/schedule_generators.py b/flatland/envs/schedule_generators.py index a8a5be77b75000900be67d2fc64dd22b46b68fae..1cbac110c2243b476db4487c534c4d886be636a8 100644 --- a/flatland/envs/schedule_generators.py +++ b/flatland/envs/schedule_generators.py @@ -154,7 +154,7 @@ def random_schedule_generator(speed_ratio_map: Mapping[float, float] = None) -> cnt = 0 while re_generate: cnt += 1 - if cnt >= 1: + if cnt > 1: print("re_generate cnt={}".format(cnt)) if cnt > 1000: raise Exception("After 1000 re_generates still not success, giving up.")