From 8fa0f57b2093fc28f6bcddf2e23241683336c5d7 Mon Sep 17 00:00:00 2001 From: sergiopaniego Date: Thu, 23 Dec 2021 17:36:09 +0100 Subject: [PATCH] Wait for experiments to be completed and saved before finishing execution --- behavior_metrics/driver.py | 2 ++ behavior_metrics/pilot.py | 1 + 2 files changed, 3 insertions(+) diff --git a/behavior_metrics/driver.py b/behavior_metrics/driver.py index 7f18bdad..7e8e2f52 100644 --- a/behavior_metrics/driver.py +++ b/behavior_metrics/driver.py @@ -195,6 +195,8 @@ def main(): logger.info('Executing app') else: script_manager.run_brains_worlds(app_configuration, controller, randomize=config_data['random']) + while not controller.pilot.execution_completed: + pass logger.info('closing all processes...') environment.close_gazebo() logger.info('DONE! Bye, bye :)') diff --git a/behavior_metrics/pilot.py b/behavior_metrics/pilot.py index 267f84e2..8c128882 100644 --- a/behavior_metrics/pilot.py +++ b/behavior_metrics/pilot.py @@ -127,6 +127,7 @@ def run(self): while not self.kill_event.is_set(): start_time = datetime.now() if not self.stop_event.is_set(): + self.execution_completed = False stopped_brain_stats = True try: self.brains.active_brain.execute()