lewtun HF staff commited on
Commit
ea714fc
1 Parent(s): 7b53738

Fix model ID sorting

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -87,7 +87,7 @@ def get_compatible_models(task: str, dataset_ids: List[str]) -> List[str]:
87
  library=["transformers", "pytorch"],
88
  )
89
  compatible_models.extend(HfApi().list_models(filter=model_filter))
90
- return set(sorted([model.modelId for model in compatible_models]))
91
 
92
 
93
  def get_key(col_mapping, val):
 
87
  library=["transformers", "pytorch"],
88
  )
89
  compatible_models.extend(HfApi().list_models(filter=model_filter))
90
+ return sorted(set([model.modelId for model in compatible_models]))
91
 
92
 
93
  def get_key(col_mapping, val):