From 94959d6ba47ba645ae06acb411c48386757c8750 Mon Sep 17 00:00:00 2001 From: Erik Nygren <erik.nygren@sbb.ch> Date: Tue, 16 Jul 2019 11:52:34 -0400 Subject: [PATCH] updated how generator check for distance map data. Updated test for generators to check that distance map is not recomputed when loaded correctly fixed formatting and removed debug prints --- flatland/envs/generators.py | 1 - 1 file changed, 1 deletion(-) diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py index ae2968f8..355f5502 100644 --- a/flatland/envs/generators.py +++ b/flatland/envs/generators.py @@ -231,7 +231,6 @@ def rail_from_file(filename): if b"distance_maps" in data.keys(): distance_maps = data[b"distance_maps"] if len(distance_maps) > 0: - print("Loading distance map") return rail, agents_position, agents_direction, agents_target, [1.0] * len( agents_position), distance_maps else: -- GitLab