idolezal commited on
Commit
e0ba174
1 Parent(s): ce9b173

Replaced RuntimeError with CheckSignificanceError

Browse files
Files changed (1) hide show
  1. server.py +7 -4
server.py CHANGED
@@ -57,6 +57,9 @@ MARKDOWN_SPECIAL_CHARACTERS = {
57
  def xmlAndMarkdownEscape(text):
58
  return xmlEscape(text, MARKDOWN_SPECIAL_CHARACTERS)
59
 
 
 
 
60
  def check_significance_send_task(model_a_path, model_b_path):
61
  url = 'https://czechllm.fit.vutbr.cz/benczechmark-leaderboard/compare_significance/'
62
 
@@ -76,9 +79,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.')
80
  else:
81
- raise RuntimeError(f'Failed to submit task. Status code: {response.status_code}')
82
 
83
  return result_url
84
 
@@ -91,12 +94,12 @@ 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}')
95
 
96
  if result["state"] == "COMPLETED":
97
  return result['result']
98
  else:
99
- raise RuntimeError(result['result']['error'])
100
 
101
  def check_significance(model_a_path, model_b_path):
102
  result_url = check_significance_send_task(model_a_path, model_b_path)
 
57
  def xmlAndMarkdownEscape(text):
58
  return xmlEscape(text, MARKDOWN_SPECIAL_CHARACTERS)
59
 
60
+ class CheckSignificanceError(Exception):
61
+ pass
62
+
63
  def check_significance_send_task(model_a_path, model_b_path):
64
  url = 'https://czechllm.fit.vutbr.cz/benczechmark-leaderboard/compare_significance/'
65
 
 
79
  result_url = response.url
80
  #task_id = response.json()['task_id']
81
  elif response.status_code == 429:
82
+ raise CheckSignificanceError('Server is too busy. Please try again later.')
83
  else:
84
+ raise CheckSignificanceError(f'Failed to submit task. Status code: {response.status_code}')
85
 
86
  return result_url
87
 
 
94
  elif response.status_code == 202:
95
  time.sleep(5)
96
  else:
97
+ raise CheckSignificanceError(f'Failed to get result. Status code: {response.status_code}')
98
 
99
  if result["state"] == "COMPLETED":
100
  return result['result']
101
  else:
102
+ raise CheckSignificanceError(result['result']['error'])
103
 
104
  def check_significance(model_a_path, model_b_path):
105
  result_url = check_significance_send_task(model_a_path, model_b_path)