fix optional

This commit is contained in:
Xi Yan 2024-11-07 16:22:33 -08:00
parent fd581c3d88
commit 7ca479f400
3 changed files with 6 additions and 6 deletions

View file

@ -212,7 +212,7 @@ class ScoringRouter(Scoring):
async def score_batch(
self,
dataset_id: str,
scoring_functions: Optional[Dict[str, ScoringFnParams]] = None,
scoring_functions: Dict[str, Optional[ScoringFnParams]] = None,
save_results_dataset: bool = False,
) -> ScoreBatchResponse:
res = {}
@ -235,7 +235,7 @@ class ScoringRouter(Scoring):
async def score(
self,
input_rows: List[Dict[str, Any]],
scoring_functions: Optional[Dict[str, ScoringFnParams]] = None,
scoring_functions: Dict[str, Optional[ScoringFnParams]] = None,
) -> ScoreResponse:
res = {}
# look up and map each scoring function to its provider impl