Merge branch 'main' into iql_comparison

facebookresearch · Sep 26, 2023 · ed0860d · ed0860d
2 parents f4530c9 + 43308cc
commit ed0860d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/benchmarl/experiment/experiment.py b/benchmarl/experiment/experiment.py
@@ -329,7 +329,12 @@ def _setup_logger(self):
     def run(self):
         try:
             self._collection_loop()
+        except KeyboardInterrupt as interrupt:
+            print("\n\nExperiment was closed gracefully\n\n")
+            self.close()
+            raise interrupt
         except Exception as err:
+            print("\n\nExperiment failed and is closing gracefully\n\n")
             self.close()
             raise err