[Feature] Add graceful experiment termination upon interrupt

Signed-off-by: Matteo Bettini <[email protected]>

benchmarl/experiment/experiment.py

@@ -322,7 +322,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