From 5de931f0b6d2befb8c511bade5a709ad83b78771 Mon Sep 17 00:00:00 2001
From: SP Mohanty <spmohanty91@gmail.com>
Date: Wed, 24 Jul 2019 13:49:17 +0200
Subject: [PATCH] Add a check to ensure that visualizations are generated only
if screenshots exist
---
flatland/evaluators/service.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/flatland/evaluators/service.py b/flatland/evaluators/service.py
index 5ae0580d..445d9226 100644
--- a/flatland/evaluators/service.py
+++ b/flatland/evaluators/service.py
@@ -146,7 +146,6 @@ class FlatlandRemoteEvaluationService:
os.path.join(root, file)
)
env_paths = sorted(env_paths)
- print(self.video_generation_envs)
for _idx, env_path in enumerate(env_paths):
"""
Here we collect the indices of the environments for which
@@ -157,7 +156,6 @@ class FlatlandRemoteEvaluationService:
env_step
"""
for vg_env in self.video_generation_envs:
- print(vg_env, env_path)
if vg_env in env_path:
self.video_generation_indices.append(_idx+1)
return sorted(env_paths)
@@ -431,7 +429,7 @@ class FlatlandRemoteEvaluationService:
mean_reward = np.mean(self.simulation_rewards)
mean_percentage_complete = np.mean(self.simulation_percentage_complete)
- if self.visualize:
+ if self.visualize and len(os.listdir(self.vizualization_folder_name)) > 0:
# Generate the video
#
# Note, if you had depdency issues due to ffmpeg, you can
--
GitLab