“WadoodAbdul” commited on
Commit
91818cc
·
1 Parent(s): 50c89b2

updated model name key

Browse files
Files changed (1) hide show
  1. src/populate.py +3 -3
src/populate.py CHANGED
@@ -34,7 +34,7 @@ def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]:
34
  with open(file_path) as fp:
35
  data = json.load(fp)
36
 
37
- data[EvalQueueColumn.model.name] = make_clickable_model(data["model"])
38
  data[EvalQueueColumn.revision.name] = data.get("revision", "main")
39
 
40
  all_evals.append(data)
@@ -45,8 +45,8 @@ def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]:
45
  file_path = os.path.join(save_path, entry, sub_entry)
46
  with open(file_path) as fp:
47
  data = json.load(fp)
48
-
49
- data[EvalQueueColumn.model.name] = make_clickable_model(data["model"])
50
  data[EvalQueueColumn.revision.name] = data.get("revision", "main")
51
  all_evals.append(data)
52
 
 
34
  with open(file_path) as fp:
35
  data = json.load(fp)
36
 
37
+ data[EvalQueueColumn.model.name] = make_clickable_model(data["model_name"])
38
  data[EvalQueueColumn.revision.name] = data.get("revision", "main")
39
 
40
  all_evals.append(data)
 
45
  file_path = os.path.join(save_path, entry, sub_entry)
46
  with open(file_path) as fp:
47
  data = json.load(fp)
48
+ # print(data)
49
+ data[EvalQueueColumn.model.name] = make_clickable_model(data["model_name"])
50
  data[EvalQueueColumn.revision.name] = data.get("revision", "main")
51
  all_evals.append(data)
52