jeqin commited on
Commit
fd9c401
·
1 Parent(s): 0d65fdc

save result when KeyboardInterrupt

Browse files
Files changed (1) hide show
  1. test_diffusers.py +2 -0
test_diffusers.py CHANGED
@@ -99,6 +99,8 @@ def test_base_model(result: list):
99
  try:
100
  ret = cmd(command, capture_output=True)
101
  result.append(parse_log(ret.stdout))
 
 
102
  except Exception as e:
103
  print("Exception: ", e)
104
  return result
 
99
  try:
100
  ret = cmd(command, capture_output=True)
101
  result.append(parse_log(ret.stdout))
102
+ except KeyboardInterrupt:
103
+ return result
104
  except Exception as e:
105
  print("Exception: ", e)
106
  return result