Unverified Commit 9bafedc0 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Remove image_transforms functions from init (#20704)

parent d994473b
......@@ -99,6 +99,7 @@ _import_structure = {
"file_utils": [],
"generation": ["GenerationConfig"],
"hf_argparser": ["HfArgumentParser"],
"image_transforms": [],
"integrations": [
"is_clearml_available",
"is_comet_available",
......@@ -748,7 +749,6 @@ except OptionalDependencyNotAvailable:
]
else:
_import_structure["image_processing_utils"] = ["ImageProcessingMixin"]
_import_structure["image_transforms"] = ["rescale", "resize", "to_pil_image"]
_import_structure["image_utils"] = ["ImageFeatureExtractionMixin"]
_import_structure["models.beit"].extend(["BeitFeatureExtractor", "BeitImageProcessor"])
_import_structure["models.bit"].extend(["BitImageProcessor"])
......@@ -3949,7 +3949,6 @@ if TYPE_CHECKING:
from .utils.dummy_vision_objects import *
else:
from .image_processing_utils import ImageProcessingMixin
from .image_transforms import rescale, resize, to_pil_image
from .image_utils import ImageFeatureExtractionMixin
from .models.beit import BeitFeatureExtractor, BeitImageProcessor
from .models.bit import BitImageProcessor
......
......@@ -10,18 +10,6 @@ class ImageProcessingMixin(metaclass=DummyObject):
requires_backends(self, ["vision"])
def rescale(*args, **kwargs):
requires_backends(rescale, ["vision"])
def resize(*args, **kwargs):
requires_backends(resize, ["vision"])
def to_pil_image(*args, **kwargs):
requires_backends(to_pil_image, ["vision"])
class ImageFeatureExtractionMixin(metaclass=DummyObject):
_backends = ["vision"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment