From 5b2546a34bc5ebb4f566eb3610eb59ddd297bf35 Mon Sep 17 00:00:00 2001 From: Celina Hanouti Date: Tue, 18 Feb 2025 18:40:36 +0100 Subject: [PATCH] simplify typing --- src/huggingface_hub/utils/_xet.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/src/huggingface_hub/utils/_xet.py b/src/huggingface_hub/utils/_xet.py index bb05e65f5a..f5628d0c0d 100644 --- a/src/huggingface_hub/utils/_xet.py +++ b/src/huggingface_hub/utils/_xet.py @@ -1,7 +1,5 @@ from dataclasses import dataclass -from typing import Dict, Optional, Union - -from requests.structures import CaseInsensitiveDict +from typing import Dict, Optional from .. import constants from . import get_session, validate_hf_hub_args @@ -16,9 +14,7 @@ class XetMetadata: file_hash: Optional[str] = None -def parse_xet_headers( - headers: Union[Dict[str, str], CaseInsensitiveDict[str]], -) -> Optional[XetMetadata]: +def parse_xet_headers(headers: Dict[str, str]) -> Optional[XetMetadata]: """ Parse XET metadata from the HTTP headers or return None if not found.