in yourbench/main.py [0:0]
def build(self) -> dict:
"""Build the configuration dictionary."""
config = {
"hf_configuration": {
"hf_dataset_name": self.hf_dataset_name,
"hf_organization": self.hf_organization,
"token": self.hf_token,
"private": self.private,
"concat_if_exist": self.concat_if_exist,
}
}
# Add local dataset options if configured
if self.local_dataset_dir:
config["hf_configuration"]["local_dataset_dir"] = self.local_dataset_dir
config["hf_configuration"]["local_saving"] = self.local_saving
config["model_list"] = self.models
if self.model_roles:
config["model_roles"] = self.model_roles
config["pipeline"] = self.pipeline_config
return config