[Fix] Gracefully close in case of error

Signed-off-by: Matteo Bettini <[email protected]>
facebookresearch · Sep 22, 2023 · d671c4d · d671c4d
1 parent 7aa4e8b
commit d671c4d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/benchmarl/experiment/experiment.py b/benchmarl/experiment/experiment.py
@@ -317,7 +317,11 @@ def _setup_logger(self):
         )
 
     def run(self):
-        self._collection_loop()
+        try:
+            self._collection_loop()
+        except Exception as err:
+            self.close()
+            raise err
 
     def _collection_loop(self):