fix typing

bespokelabsai · Dec 13, 2024 · 19089cb · 19089cb
1 parent 82817e5
commit 19089cb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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,
             ]