Include catching ctrl-C

facebookresearch · Jun 27, 2023 · 13ad2af · 13ad2af
1 parent 4f72d05
commit 13ad2af
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mephisto/abstractions/architects/ec2/ec2_helpers.py b/mephisto/abstractions/architects/ec2/ec2_helpers.py
@@ -969,7 +969,7 @@ def deploy_fallback_server(
             env=dict(os.environ, SSH_AUTH_SOCK=""),
         )
         detete_instance_address(session, allocation_id, association_id)
-    except Exception as e:
+    except (Exception, KeyboardInterrupt) as e:
         old_handler = signal.signal(signal.SIGINT, cant_cancel_shutdown)
         detete_instance_address(session, allocation_id, association_id)
         signal.signal(signal.SIGINT, old_handler)
@@ -1020,7 +1020,7 @@ def deploy_to_routing_server(
         )
         detete_instance_address(session, allocation_id, association_id)
         print("Server setup complete!")
-    except Exception as e:
+    except (Exception, KeyboardInterrupt) as e:
         old_handler = signal.signal(signal.SIGINT, cant_cancel_shutdown)
         detete_instance_address(session, allocation_id, association_id)
         signal.signal(signal.SIGINT, old_handler)