From 0be360158a0768c9e1bf95b076a577a2109bda74 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 17 Sep 2019 11:11:39 +0200
Subject: [PATCH] refactoring and clean up
---
examples/Simple_Realistic_Railway_Generator.py | 11 ++++++-----
1 file changed, 6 insertions(+), 5 deletions(-)
diff --git a/examples/Simple_Realistic_Railway_Generator.py b/examples/Simple_Realistic_Railway_Generator.py
index 51703606..732819d6 100644
--- a/examples/Simple_Realistic_Railway_Generator.py
+++ b/examples/Simple_Realistic_Railway_Generator.py
@@ -476,10 +476,11 @@ for itrials in range(1000):
env_renderer.render_env(show=True, show_observations=False, show_predictions=False)
cnt += 1
- env_renderer.gl.save_image(
- os.path.join(
- "./../render_output/",
- "flatland_frame_{:04d}_{:04d}.png".format(itrials, 0)
- ))
+ if os.path.exists("./../render_output/"):
+ env_renderer.gl.save_image(
+ os.path.join(
+ "./../render_output/",
+ "flatland_frame_{:04d}_{:04d}.png".format(itrials, 0)
+ ))
env_renderer.close_window()
--
GitLab