KennethEnevoldsen commited on
Commit
c8e4be5
1 Parent(s): 9c65fce

added fix to paws

Browse files
Files changed (1) hide show
  1. refresh.py +5 -2
refresh.py CHANGED
@@ -214,8 +214,8 @@ def get_mteb_data(tasks=["Clustering"], langs=[], datasets=[], fillna=True, add_
214
  datasets.append('MLSUMClusteringP2P')
215
  if ('MLSUMClusteringS2S (fr)' in datasets):
216
  datasets.append('MLSUMClusteringS2S')
217
- if ('PawsX (fr)' in datasets):
218
- datasets.append('PawsXPairClassification (fr)')
219
  # Initialize list to models that we cannot fetch metadata from
220
  df_list = []
221
  for model in external_model_results:
@@ -322,6 +322,9 @@ def get_mteb_data(tasks=["Clustering"], langs=[], datasets=[], fillna=True, add_
322
  if ('MLSUMClusteringS2S (fr)' in datasets) and ('MLSUMClusteringS2S' in cols):
323
  df['MLSUMClusteringS2S (fr)'] = df['MLSUMClusteringS2S (fr)'].fillna(df['MLSUMClusteringS2S'])
324
  datasets.remove('MLSUMClusteringS2S')
 
 
 
325
  # Filter invalid columns
326
  cols = [col for col in cols if col in base_columns + datasets]
327
  i = 0
 
214
  datasets.append('MLSUMClusteringP2P')
215
  if ('MLSUMClusteringS2S (fr)' in datasets):
216
  datasets.append('MLSUMClusteringS2S')
217
+ if ('PawsXPairClassification (fr)' in datasets):
218
+ datasets.append('PawsX (fr)')
219
  # Initialize list to models that we cannot fetch metadata from
220
  df_list = []
221
  for model in external_model_results:
 
322
  if ('MLSUMClusteringS2S (fr)' in datasets) and ('MLSUMClusteringS2S' in cols):
323
  df['MLSUMClusteringS2S (fr)'] = df['MLSUMClusteringS2S (fr)'].fillna(df['MLSUMClusteringS2S'])
324
  datasets.remove('MLSUMClusteringS2S')
325
+ if ('PawsXPairClassification (fr)' in datasets) and ('PawsX (fr)' in cols):
326
+ df['PawsXPairClassification (fr)'] = df['PawsXPairClassification (fr)'].fillna(df['PawsX (fr)'])
327
+ datasets.remove('PawsX (fr)')
328
  # Filter invalid columns
329
  cols = [col for col in cols if col in base_columns + datasets]
330
  i = 0