Alina Lozovskaia commited on
Commit
d8bf61b
1 Parent(s): 5a8f7dc

small fixed

Browse files
Files changed (3) hide show
  1. app.py +0 -1
  2. src/populate.py +1 -8
  3. src/tools/collections.py +1 -1
app.py CHANGED
@@ -77,7 +77,6 @@ def download_dataset(repo_id, local_dir, repo_type="dataset", max_attempts=3):
77
  attempt += 1
78
  if attempt == max_attempts:
79
  restart_space()
80
- break
81
 
82
 
83
  def init_space(full_init: bool = True):
 
77
  attempt += 1
78
  if attempt == max_attempts:
79
  restart_space()
 
80
 
81
 
82
  def init_space(full_init: bool = True):
src/populate.py CHANGED
@@ -21,14 +21,7 @@ def get_evaluation_queue_df(save_path, cols):
21
  save_path = pathlib.Path(save_path)
22
  all_evals = []
23
 
24
- for path in save_path.rglob('*'):
25
- if path.is_dir():
26
- continue
27
- if path.name.startswith('.'):
28
- continue
29
- if path.name.endswith('.md'):
30
- continue
31
-
32
  data = load_json_data(path)
33
  if data:
34
  all_evals.append(_process_model_data(data))
 
21
  save_path = pathlib.Path(save_path)
22
  all_evals = []
23
 
24
+ for path in save_path.rglob('*.json'):
 
 
 
 
 
 
 
25
  data = load_json_data(path)
26
  if data:
27
  all_evals.append(_process_model_data(data))
src/tools/collections.py CHANGED
@@ -62,7 +62,7 @@ def update_collections(df: DataFrame):
62
  best_models = list(
63
  filtered_df.sort_values(AutoEvalColumn.average.name, ascending=False)[AutoEvalColumn.dummy.name][:10]
64
  )
65
- print(model_type.value.symbol, size, best_models[:10])
66
  _add_models_to_collection(collection, best_models, model_type, size)
67
  cur_best_models.extend(best_models)
68
 
 
62
  best_models = list(
63
  filtered_df.sort_values(AutoEvalColumn.average.name, ascending=False)[AutoEvalColumn.dummy.name][:10]
64
  )
65
+ print(model_type.value.symbol, size, best_models)
66
  _add_models_to_collection(collection, best_models, model_type, size)
67
  cur_best_models.extend(best_models)
68