Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
KennethEnevoldsen
commited on
Merge pull request #5 from embeddings-benchmark/fix-paws
Browse files- 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 ('
|
218 |
-
datasets.append('
|
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
|