From 629063cc0f970c219493ff7a7001afd715e20c36 Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Tue, 24 Sep 2019 09:35:05 +0200
Subject: [PATCH] bugfix create existing folder
---
examples/simple_example_city_railway_generator.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/examples/simple_example_city_railway_generator.py b/examples/simple_example_city_railway_generator.py
index ce96932c..4182bc27 100644
--- a/examples/simple_example_city_railway_generator.py
+++ b/examples/simple_example_city_railway_generator.py
@@ -9,7 +9,9 @@ from flatland.envs.rail_generators_city_generator import city_generator
from flatland.envs.schedule_generators import city_schedule_generator
from flatland.utils.rendertools import RenderTool, AgentRenderVariant
-os.mkdir("./../render_output/")
+OUTPUT_DIR = "./../render_output/"
+if not os.path.exists(OUTPUT_DIR):
+ os.mkdir(OUTPUT_DIR)
for itrials in np.arange(1, 15, 1):
print(itrials, "generate new city")
@@ -52,7 +54,7 @@ for itrials in np.arange(1, 15, 1):
# store rendered file into render_output if the path exists
env_renderer.gl.save_image(
os.path.join(
- "./../render_output/",
+ OUTPUT_DIR,
"flatland_frame_{:04d}.png".format(itrials)
))
--
GitLab