ThomasSimonini HF staff commited on
Commit
59ab830
1 Parent(s): 457d9a8
Files changed (1) hide show
  1. background_task.py +1 -1
background_task.py CHANGED
@@ -200,7 +200,7 @@ def get_models_list(filter_bad_models) -> list:
200
  models = []
201
  models_ids = []
202
  data = pd.read_csv(os.path.join(DATASET_REPO_URL, "resolve", "main", ELO_FILENAME))
203
- models_on_hub = api.list_models(filter=["reinforcement-learning", "ml-agents", "ML-Agents-SoccerTwos"])
204
  for i, row in data.iterrows():
205
  model_id = row["author"] + "/" + row["model"]
206
  if model_id in filter_bad_models:
 
200
  models = []
201
  models_ids = []
202
  data = pd.read_csv(os.path.join(DATASET_REPO_URL, "resolve", "main", ELO_FILENAME))
203
+ models_on_hub = api.list_models(filter=["reinforcement-learning", "ml-agents", "ML-Agents-SoccerTwos", "onnx"])
204
  for i, row in data.iterrows():
205
  model_id = row["author"] + "/" + row["model"]
206
  if model_id in filter_bad_models: