diff --git a/lang_agent/eval/evaluator.py b/lang_agent/eval/evaluator.py index aa3d17e..9994d46 100644 --- a/lang_agent/eval/evaluator.py +++ b/lang_agent/eval/evaluator.py @@ -38,9 +38,7 @@ class Evaluator: logger.info("preparing to run experiment") self.pipeline:Pipeline = self.config.pipe_config.setup() self.cli = Client() - self.validator:Validator = self.config.validator_config.setup( - dataset_name=self.config.dataset_name - ) + self.validator:Validator = self.config.validator_config.setup() self.dataset = self.cli.read_dataset(dataset_name=self.config.dataset_name) diff --git a/lang_agent/eval/validator.py b/lang_agent/eval/validator.py index b672298..3290708 100644 --- a/lang_agent/eval/validator.py +++ b/lang_agent/eval/validator.py @@ -14,9 +14,8 @@ class ValidatorConfig(KeyConfig): class Validator: - def __init__(self, config: ValidatorConfig, dataset_name:str): + def __init__(self, config: ValidatorConfig): self.config = config - self.dataset_name = dataset_name self.populate_modules()