Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

return default if not specified

Browse Source
This commit is contained in:
goulustis
2025-10-27 15:21:12 +08:00
parent b3f35d5319
commit 880a573c42
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
lang_agent/eval/validator.py
View File

@@ -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]
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)