MohamedRashad commited on
Commit
bc40b2e
Β·
1 Parent(s): 542dc0f

Normalize precision comparison to capitalize in model submission checks

Browse files
Files changed (1) hide show
  1. utils.py +3 -3
utils.py CHANGED
@@ -95,7 +95,7 @@ def submit_model(model_name, revision, precision, params, license, task):
95
  # Check if model is already evaluated
96
  model_exists_in_results = ((existing_models_results['Model'] == model_name) &
97
  (existing_models_results['Revision'] == revision) &
98
- (existing_models_results['Precision'].lower() == precision) &
99
  (existing_models_results['Task'] == task)).any()
100
  if model_exists_in_results:
101
  return f"Model {model_name} has already been evaluated as a {task} πŸŽ‰"
@@ -105,7 +105,7 @@ def submit_model(model_name, revision, precision, params, license, task):
105
  existing_models_pending = df_pending[['model_name', 'revision', 'precision', 'task']]
106
  model_exists_in_pending = ((existing_models_pending['model_name'] == model_name) &
107
  (existing_models_pending['revision'] == revision) &
108
- (existing_models_pending['precision'].lower() == precision) &
109
  (existing_models_pending['task'] == task)).any()
110
  if model_exists_in_pending:
111
  return f"Model {model_name} is already in the evaluation queue as a {task} πŸš€"
@@ -115,7 +115,7 @@ def submit_model(model_name, revision, precision, params, license, task):
115
  existing_models_finished = df_finished[['model_name', 'revision', 'precision', 'task']]
116
  model_exists_in_finished = ((existing_models_finished['model_name'] == model_name) &
117
  (existing_models_finished['revision'] == revision) &
118
- (existing_models_finished['precision'].lower() == precision) &
119
  (existing_models_finished['task'] == task)).any()
120
  if model_exists_in_finished:
121
  return f"Model {model_name} has already been evaluated as a {task} πŸŽ‰"
 
95
  # Check if model is already evaluated
96
  model_exists_in_results = ((existing_models_results['Model'] == model_name) &
97
  (existing_models_results['Revision'] == revision) &
98
+ (existing_models_results['Precision'] == precision.capitalize()) &
99
  (existing_models_results['Task'] == task)).any()
100
  if model_exists_in_results:
101
  return f"Model {model_name} has already been evaluated as a {task} πŸŽ‰"
 
105
  existing_models_pending = df_pending[['model_name', 'revision', 'precision', 'task']]
106
  model_exists_in_pending = ((existing_models_pending['model_name'] == model_name) &
107
  (existing_models_pending['revision'] == revision) &
108
+ (existing_models_pending['precision'] == precision.capitalize()) &
109
  (existing_models_pending['task'] == task)).any()
110
  if model_exists_in_pending:
111
  return f"Model {model_name} is already in the evaluation queue as a {task} πŸš€"
 
115
  existing_models_finished = df_finished[['model_name', 'revision', 'precision', 'task']]
116
  model_exists_in_finished = ((existing_models_finished['model_name'] == model_name) &
117
  (existing_models_finished['revision'] == revision) &
118
+ (existing_models_finished['precision'] == precision.capitalize()) &
119
  (existing_models_finished['task'] == task)).any()
120
  if model_exists_in_finished:
121
  return f"Model {model_name} has already been evaluated as a {task} πŸŽ‰"