From a057b2aed9aeb1fe141975e87818820a67b577d1 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 20 Aug 2019 10:53:28 +0200
Subject: [PATCH] rendering error
---
examples/flatland_2_0_example.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py
index 1caa0aa9..5078af15 100644
--- a/examples/flatland_2_0_example.py
+++ b/examples/flatland_2_0_example.py
@@ -18,7 +18,6 @@ stochastic_data = {'prop_malfunction': 0.5, # Percentage of defective agents
'max_duration': 10 # Max duration of malfunction
}
-
TreeObservation = TreeObsForRailEnv(max_depth=2, predictor=ShortestPathPredictorForRailEnv())
env = RailEnv(width=10,
height=10,
@@ -106,7 +105,10 @@ for trials in range(1, n_trials + 1):
# reward and whether their are done
next_obs, all_rewards, done, _ = env.step(action_dict)
env_renderer.render_env(show=True, show_observations=False, show_predictions=False)
- env_renderer.gl.save_image("./Images/flatland_2_0_frame_{:04d}.bmp".format(frame_step))
+ try:
+ env_renderer.gl.save_image("./../rendering/flatland_2_0_frame_{:04d}.bmp".format(frame_step))
+ except:
+ print("Path not found: ./../rendering/")
frame_step += 1
# Update replay buffer and train agent
for a in range(env.get_num_agents()):
--
GitLab