XufengDuan commited on
Commit
54e251e
1 Parent(s): 1bde8d8

update scripts

Browse files
Files changed (2) hide show
  1. src/display/utils.py +4 -4
  2. src/populate.py +2 -2
src/display/utils.py CHANGED
@@ -57,10 +57,10 @@ AutoEvalColumn = make_dataclass("AutoEvalColumn", auto_eval_column_dict, frozen=
57
  @dataclass(frozen=True)
58
  class EvalQueueColumn: # Queue column
59
  model = ColumnContent("model", "markdown", True)
60
- revision = ColumnContent("revision", "str", True)
61
- private = ColumnContent("private", "bool", True)
62
- precision = ColumnContent("precision", "str", True)
63
- weight_type = ColumnContent("weight_type", "str", "Original")
64
  status = ColumnContent("status", "str", True)
65
 
66
  ## All the model information that we might need
 
57
  @dataclass(frozen=True)
58
  class EvalQueueColumn: # Queue column
59
  model = ColumnContent("model", "markdown", True)
60
+ # revision = ColumnContent("revision", "str", True)
61
+ # private = ColumnContent("private", "bool", True)
62
+ # precision = ColumnContent("precision", "str", True)
63
+ # weight_type = ColumnContent("weight_type", "str", "Original")
64
  status = ColumnContent("status", "str", True)
65
 
66
  ## All the model information that we might need
src/populate.py CHANGED
@@ -40,7 +40,7 @@ def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]:
40
  data = json.load(fp)
41
 
42
  data[utils.EvalQueueColumn.model.name] = data["model"] #formatting.make_clickable_model(data["model"])
43
- data[utils.EvalQueueColumn.revision.name] = data.get("revision", "main")
44
 
45
  all_evals.append(data)
46
  elif ".md" not in entry:
@@ -52,7 +52,7 @@ def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]:
52
  data = json.load(fp)
53
 
54
  data[utils.EvalQueueColumn.model.name] = formatting.make_clickable_model(data["model"])
55
- data[utils.EvalQueueColumn.revision.name] = data.get("revision", "main")
56
  all_evals.append(data)
57
 
58
  pending_list = [e for e in all_evals if e["status"] in ["PENDING", "RERUN"]]
 
40
  data = json.load(fp)
41
 
42
  data[utils.EvalQueueColumn.model.name] = data["model"] #formatting.make_clickable_model(data["model"])
43
+ # data[utils.EvalQueueColumn.revision.name] = data.get("revision", "main")
44
 
45
  all_evals.append(data)
46
  elif ".md" not in entry:
 
52
  data = json.load(fp)
53
 
54
  data[utils.EvalQueueColumn.model.name] = formatting.make_clickable_model(data["model"])
55
+ # data[utils.EvalQueueColumn.revision.name] = data.get("revision", "main")
56
  all_evals.append(data)
57
 
58
  pending_list = [e for e in all_evals if e["status"] in ["PENDING", "RERUN"]]