wehub-resource-sync
16 lines
531 B
Python
16 lines
531 B
Python
from FlagEmbedding.abc.evaluation import AbsEvalRunner
|
|
|
|
from .data_loader import CustomEvalDataLoader
|
|
|
|
|
|
class CustomEvalRunner(AbsEvalRunner):
|
|
def load_data_loader(self) -> CustomEvalDataLoader:
|
|
data_loader = CustomEvalDataLoader(
|
|
eval_name=self.eval_args.eval_name,
|
|
dataset_dir=self.eval_args.dataset_dir,
|
|
cache_dir=self.eval_args.cache_path,
|
|
token=self.eval_args.token,
|
|
force_redownload=self.eval_args.force_redownload,
|
|
)
|
|
return data_loader
|