mypy

src/huggingface_hub/_webhooks_payload.py

@@ -14,7 +14,7 @@
 # limitations under the License.
 """Contains data structures to parse the webhooks payload."""
 
-from typing import Any, List, Literal, Optional
+from typing import Any, List, Literal, Optional, Union
 
 from .utils import is_pydantic_available
 
@@ -47,14 +47,14 @@ def schema(cls, *args, **kwargs) -> dict[str, Any]:
             )
 
         @classmethod
-        def model_validate_json(cls, json_data: str | bytes | bytearray, *args, **kwargs) -> "BaseModel":
+        def model_validate_json(cls, json_data: Union[str, bytes, bytearray], *args, **kwargs) -> "BaseModel":
             raise ImportError(
                 "You must have `pydantic` installed to use `WebhookPayload`. This is an optional dependency that"
                 " should be installed separately. Please run `pip install --upgrade pydantic` and retry."
             )
 
         @classmethod
-        def parse_raw(cls, json_data: str | bytes | bytearray, *args, **kwargs) -> "BaseModel":
+        def parse_raw(cls, json_data: Union[str, bytes, bytearray], *args, **kwargs) -> "BaseModel":
             raise ImportError(
                 "You must have `pydantic` installed to use `WebhookPayload`. This is an optional dependency that"
                 " should be installed separately. Please run `pip install --upgrade pydantic` and retry."

src/huggingface_hub/inference/_client.py

@@ -877,7 +877,7 @@ def chat_completion(
         ActivitySummary(location='park', activity='bike ride', animals_seen=3, animals=['puppy', 'cat', 'raccoon'])
         ```
         """
-        if issubclass(response_format, BaseModel):
+        if isinstance(response_format, type) and issubclass(response_format, BaseModel):
             response_model = response_format
             response_format = ChatCompletionInputGrammarType(
                 type="json",

src/huggingface_hub/inference/_generated/_async_client.py

@@ -932,7 +932,7 @@ async def chat_completion(
         ActivitySummary(location='park', activity='bike ride', animals_seen=3, animals=['puppy', 'cat', 'raccoon'])
         ```
         """
-        if issubclass(response_format, BaseModel):
+        if isinstance(response_format, type) and issubclass(response_format, BaseModel):
             response_model = response_format
             response_format = ChatCompletionInputGrammarType(
                 type="json",