From dee492d36b13de4c7560e0e339529799b4806cd9 Mon Sep 17 00:00:00 2001
From: Chandan Singh <chandan_singh@berkeley.edu>
Date: Wed, 1 Feb 2023 09:43:45 -0800
Subject: [PATCH] add easy interrupting sweeps

---
 scripts/submit_utils.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/scripts/submit_utils.py b/scripts/submit_utils.py
index 2a06a28..45463f3 100644
--- a/scripts/submit_utils.py
+++ b/scripts/submit_utils.py
@@ -1,10 +1,10 @@
 from typing import Any, Dict, List, Tuple
 
 import itertools
-import os
+import subprocess
 import random
 from functools import reduce
-"""Handles utilities for submission.
+"""Handles utilities for job sweeps.
 This file probably does not need to be edited.
 """
 
@@ -50,7 +50,11 @@ def run_args_list(
             f'\n\n-------------------{i + 1}/{len(args_list)}--------------------\n', param_str)
         try:
             if actually_run:
-                os.system(param_str)
+                # os.system(param_str)
+                sts = subprocess.Popen(param_str, shell=True).wait()
+        except KeyboardInterrupt:
+            print('Keyboard interrupt, exiting...')
+            exit(0)
         except Exception as e:
             print(e)
 
@@ -116,3 +120,4 @@ def _validate_arguments(
         for k in k_tup:
             assert not k in params_shared_dict, f"params_coupled_dict key {k} should not be in params_shared_dict"
 
+