From 880a573c428b5018f32a2552d107c58b08a6dd34 Mon Sep 17 00:00:00 2001 From: goulustis Date: Mon, 27 Oct 2025 15:21:12 +0800 Subject: [PATCH] return default if not specified --- lang_agent/eval/validator.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/lang_agent/eval/validator.py b/lang_agent/eval/validator.py index 47eeedb..614ae80 100644 --- a/lang_agent/eval/validator.py +++ b/lang_agent/eval/validator.py @@ -1,5 +1,5 @@ from dataclasses import dataclass, field -from typing import Type, Literal +from typing import Type, Callable, List import tyro from lang_agent.config import KeyConfig @@ -71,9 +71,10 @@ class Validator: return pipeline.chat(inp, as_raw=True) - def get_val_fnc(self, dataset_name:str): - return self.dict_corr_map[dataset_name] + def get_val_fnc(self, dataset_name:str)->List[Callable]: + return self.dict_corr_map.get(dataset_name, [self.Toxic_Queries_correct]) - def get_inp_fnc(self,dataset_name:str): - return self.dict_inp_map[dataset_name] \ No newline at end of file + def get_inp_fnc(self,dataset_name:str)->Callable: + # return self.dict_inp_map[dataset_name] + return self.dict_corr_map.get(dataset_name, self.Toxic_Queries_inp_parse) \ No newline at end of file