Clean up Unpack imports (#33631)

clean up Unpack imports
This commit is contained in:
Pablo Montalvo
2024-09-23 10:21:17 +02:00
committed by GitHub
parent 78b2929c05
commit 9eb93854b9
6 changed files with 6 additions and 44 deletions

View File

@@ -17,20 +17,12 @@ Processor class for Grounding DINO.
"""
import pathlib
import sys
from typing import Dict, List, Optional, Tuple, Union
from ...image_processing_utils import BatchFeature
from ...image_transforms import center_to_corners_format
from ...image_utils import AnnotationFormat, ImageInput
from ...processing_utils import ImagesKwargs, ProcessingKwargs, ProcessorMixin
if sys.version_info >= (3, 11):
from typing import Unpack
else:
from typing_extensions import Unpack
from ...processing_utils import ImagesKwargs, ProcessingKwargs, ProcessorMixin, Unpack
from ...tokenization_utils_base import BatchEncoding, PreTokenizedInput, TextInput
from ...utils import TensorType, is_torch_available

View File

@@ -16,21 +16,15 @@
Processor class for Llava.
"""
import sys
from typing import List, Union
from ...feature_extraction_utils import BatchFeature
from ...image_utils import ImageInput, get_image_size, to_numpy_array
from ...processing_utils import ProcessingKwargs, ProcessorMixin, _validate_images_text_input_order
from ...processing_utils import ProcessingKwargs, ProcessorMixin, Unpack, _validate_images_text_input_order
from ...tokenization_utils_base import PreTokenizedInput, TextInput
from ...utils import logging
if sys.version_info >= (3, 11):
from typing import Unpack
else:
from typing_extensions import Unpack
logger = logging.get_logger(__name__)

View File

@@ -18,22 +18,12 @@ Processor class for LLaVa-Onevision.
import math
import os
import sys
from typing import Iterable, List, Union
if sys.version_info >= (3, 11):
from typing import Unpack
else:
from typing_extensions import Unpack
from ...feature_extraction_utils import BatchFeature
from ...image_processing_utils import select_best_resolution
from ...image_utils import ImageInput, VideoInput, get_image_size, to_numpy_array
from ...processing_utils import (
ProcessingKwargs,
ProcessorMixin,
)
from ...processing_utils import ProcessingKwargs, ProcessorMixin, Unpack
from ...tokenization_utils_base import PreTokenizedInput, TextInput
from ...utils import logging
from ..auto import AutoImageProcessor

View File

@@ -16,21 +16,15 @@
Processor class for Pixtral.
"""
import sys
from typing import List, Union
from ...feature_extraction_utils import BatchFeature
from ...image_utils import ImageInput, is_valid_image, load_image
from ...processing_utils import ProcessingKwargs, ProcessorMixin, _validate_images_text_input_order
from ...processing_utils import ProcessingKwargs, ProcessorMixin, Unpack, _validate_images_text_input_order
from ...tokenization_utils_base import PreTokenizedInput, TextInput
from ...utils import is_torch_device, is_torch_dtype, is_torch_tensor, logging, requires_backends
if sys.version_info >= (3, 11):
from typing import Unpack
else:
from typing_extensions import Unpack
logger = logging.get_logger(__name__)

View File

@@ -23,18 +23,9 @@ Processor class for Qwen2-VL.
from typing import List, Union
try:
from typing import Unpack
except ImportError:
from typing_extensions import Unpack
from ...feature_extraction_utils import BatchFeature
from ...image_utils import ImageInput, VideoInput
from ...processing_utils import (
ProcessingKwargs,
ProcessorMixin,
)
from ...processing_utils import ProcessingKwargs, ProcessorMixin, Unpack
from ...tokenization_utils_base import PreTokenizedInput, TextInput
from ...utils import logging

View File

@@ -21,6 +21,7 @@ import tempfile
import numpy as np
from transformers.models.auto.processing_auto import processor_class_from_name
from transformers.processing_utils import Unpack
from transformers.testing_utils import (
check_json_file_has_correct_format,
require_torch,