eduagarcia commited on
Commit
e2369db
1 Parent(s): 67c432c

fix enum value

Browse files
Files changed (1) hide show
  1. src/display/utils.py +2 -2
src/display/utils.py CHANGED
@@ -263,8 +263,8 @@ if os.path.exists('external_models_results.json'):
263
  model_row[AutoEvalColumn.npm.name] = round(model_data['result_metrics_npm']*100, 2)
264
 
265
  model_type = ModelType.from_str(model_data['model_type'])
266
- model_row[AutoEvalColumn.model_type.name] = model_type.name
267
- model_row[AutoEvalColumn.model_type_symbol.name] = model_type.symbol
268
  if model_type == ModelType.proprietary:
269
  model_row[AutoEvalColumn.license.name] = "Proprietary"
270
  if 'params' in model_data:
 
263
  model_row[AutoEvalColumn.npm.name] = round(model_data['result_metrics_npm']*100, 2)
264
 
265
  model_type = ModelType.from_str(model_data['model_type'])
266
+ model_row[AutoEvalColumn.model_type.name] = model_type.value.name
267
+ model_row[AutoEvalColumn.model_type_symbol.name] = model_type.value.symbol
268
  if model_type == ModelType.proprietary:
269
  model_row[AutoEvalColumn.license.name] = "Proprietary"
270
  if 'params' in model_data: