idolezal commited on
Commit
ce9b173
ยท
1 Parent(s): 8d98c0c

Exception gr.error doesn't make sense here

Browse files
Files changed (1) hide show
  1. server.py +3 -3
server.py CHANGED
@@ -76,9 +76,9 @@ def check_significance_send_task(model_a_path, model_b_path):
76
  result_url = response.url
77
  #task_id = response.json()['task_id']
78
  elif response.status_code == 429:
79
- raise RuntimeError('Server is too busy. Please try again later.') # TODO: try-except do raise gr.error
80
  else:
81
- raise RuntimeError(f'Failed to submit task. Status code: {response.status_code}') # TODO: try-except do raise gr.error
82
 
83
  return result_url
84
 
@@ -91,7 +91,7 @@ def check_significance_wait_for_result(result_url):
91
  elif response.status_code == 202:
92
  time.sleep(5)
93
  else:
94
- raise RuntimeError(f'Failed to get result. Status code: {response.status_code}') # TODO: try-except do raise gr.error
95
 
96
  if result["state"] == "COMPLETED":
97
  return result['result']
 
76
  result_url = response.url
77
  #task_id = response.json()['task_id']
78
  elif response.status_code == 429:
79
+ raise RuntimeError('Server is too busy. Please try again later.')
80
  else:
81
+ raise RuntimeError(f'Failed to submit task. Status code: {response.status_code}')
82
 
83
  return result_url
84
 
 
91
  elif response.status_code == 202:
92
  time.sleep(5)
93
  else:
94
+ raise RuntimeError(f'Failed to get result. Status code: {response.status_code}')
95
 
96
  if result["state"] == "COMPLETED":
97
  return result['result']