From 19089cb2e9c1cbbba876ff4629884c025137057c Mon Sep 17 00:00:00 2001 From: Ryan Marten Date: Fri, 13 Dec 2024 14:43:21 -0800 Subject: [PATCH] fix typing --- src/bespokelabs/curator/llm/llm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/bespokelabs/curator/llm/llm.py b/src/bespokelabs/curator/llm/llm.py index 08bad431..589b3e4f 100644 --- a/src/bespokelabs/curator/llm/llm.py +++ b/src/bespokelabs/curator/llm/llm.py @@ -44,8 +44,8 @@ def __init__( parse_func: Optional[ Callable[ [ - Union[Dict[str, Any], BaseModel], - Union[Dict[str, Any], BaseModel], + _DictOrBaseModel, + _DictOrBaseModel, ], T, ]