From d30e29ad9d74cdb1e692c2b059e12a9f5b304fdf Mon Sep 17 00:00:00 2001 From: EdanToledo Date: Tue, 27 Aug 2024 21:12:51 +0000 Subject: [PATCH] fix: clear pipeline before joining actors - fixes deadlock issue --- stoix/systems/ppo/sebulba/ff_ppo.py | 1 + 1 file changed, 1 insertion(+) diff --git a/stoix/systems/ppo/sebulba/ff_ppo.py b/stoix/systems/ppo/sebulba/ff_ppo.py index 200fd3d..0910b0e 100644 --- a/stoix/systems/ppo/sebulba/ff_ppo.py +++ b/stoix/systems/ppo/sebulba/ff_ppo.py @@ -807,6 +807,7 @@ def run_experiment(_config: DictConfig) -> float: # Now we stop the actors and params sources print(f"{Fore.MAGENTA}{Style.BRIGHT}Closing actors...{Style.RESET_ALL}") + pipeline.clear() for actor in actor_threads: # We clear the pipeline before stopping each actor thread # since actors can be blocked on the pipeline