Skip to content

Commit 8c3e1eb

Browse files
committed
fix bugs in data_loader.py for evaluation
- missed parameters: dataset_name
1 parent f3f9800 commit 8c3e1eb

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

FlagEmbedding/abc/evaluation/data_loader.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -266,7 +266,7 @@ def _load_local_qrels(self, save_dir: str, dataset_name: Optional[str] = None, s
266266
Returns:
267267
datasets.DatasetDict: A dict of relevance of query and document.
268268
"""
269-
checked_split = self.check_splits(split)
269+
checked_split = self.check_splits(split, dataset_name=dataset_name)
270270
if len(checked_split) == 0:
271271
raise ValueError(f"Split {split} not found in the dataset.")
272272
split = checked_split[0]
@@ -301,7 +301,7 @@ def _load_local_queries(self, save_dir: str, dataset_name: Optional[str] = None,
301301
Returns:
302302
datasets.DatasetDict: A dict of queries with id as key, query text as value.
303303
"""
304-
checked_split = self.check_splits(split)
304+
checked_split = self.check_splits(split, dataset_name=dataset_name)
305305
if len(checked_split) == 0:
306306
raise ValueError(f"Split {split} not found in the dataset.")
307307
split = checked_split[0]

0 commit comments

Comments
 (0)