hSterz commited on
Commit
708fbc1
·
1 Parent(s): cc1ef8b
Files changed (2) hide show
  1. src/display/utils.py +2 -3
  2. src/populate.py +2 -0
src/display/utils.py CHANGED
@@ -47,9 +47,8 @@ AutoEvalColumn = make_dataclass("AutoEvalColumn", auto_eval_column_dict, frozen=
47
  class EvalQueueColumn: # Queue column
48
  model = ColumnContent("model", "markdown", True)
49
  revision = ColumnContent("revision", "str", True)
50
- private = ColumnContent("private", "bool", True)
51
- precision = ColumnContent("precision", "str", True)
52
- weight_type = ColumnContent("weight_type", "str", "Original")
53
  status = ColumnContent("status", "str", True)
54
 
55
  ## All the model information that we might need
 
47
  class EvalQueueColumn: # Queue column
48
  model = ColumnContent("model", "markdown", True)
49
  revision = ColumnContent("revision", "str", True)
50
+ output_format = ColumnContent("output_format", "str", True)
51
+ dataset_version = ColumnContent("dataset_version", "str", True)
 
52
  status = ColumnContent("status", "str", True)
53
 
54
  ## All the model information that we might need
src/populate.py CHANGED
@@ -52,6 +52,8 @@ def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]:
52
 
53
  data[EvalQueueColumn.model.name] = make_clickable_model(data["model"])
54
  data[EvalQueueColumn.revision.name] = data.get("revision", "main")
 
 
55
 
56
  all_evals.append(data)
57
 
 
52
 
53
  data[EvalQueueColumn.model.name] = make_clickable_model(data["model"])
54
  data[EvalQueueColumn.revision.name] = data.get("revision", "main")
55
+ data[EvalQueueColumn.output_format] = data["output_format"]
56
+ data[EvalQueueColumn.dataset_version] = file_path.split("/")[0]
57
 
58
  all_evals.append(data)
59