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 = { ...@@ -99,6 +99,7 @@ _import_structure = {
"file_utils": [], "file_utils": [],
"generation": ["GenerationConfig"], "generation": ["GenerationConfig"],
"hf_argparser": ["HfArgumentParser"], "hf_argparser": ["HfArgumentParser"],
"image_transforms": [],
"integrations": [ "integrations": [
"is_clearml_available", "is_clearml_available",
"is_comet_available", "is_comet_available",
...@@ -748,7 +749,6 @@ except OptionalDependencyNotAvailable: ...@@ -748,7 +749,6 @@ except OptionalDependencyNotAvailable:
] ]
else: else:
_import_structure["image_processing_utils"] = ["ImageProcessingMixin"] _import_structure["image_processing_utils"] = ["ImageProcessingMixin"]
_import_structure["image_transforms"] = ["rescale", "resize", "to_pil_image"]
_import_structure["image_utils"] = ["ImageFeatureExtractionMixin"] _import_structure["image_utils"] = ["ImageFeatureExtractionMixin"]
_import_structure["models.beit"].extend(["BeitFeatureExtractor", "BeitImageProcessor"]) _import_structure["models.beit"].extend(["BeitFeatureExtractor", "BeitImageProcessor"])
_import_structure["models.bit"].extend(["BitImageProcessor"]) _import_structure["models.bit"].extend(["BitImageProcessor"])
...@@ -3949,7 +3949,6 @@ if TYPE_CHECKING: ...@@ -3949,7 +3949,6 @@ if TYPE_CHECKING:
from .utils.dummy_vision_objects import * from .utils.dummy_vision_objects import *
else: else:
from .image_processing_utils import ImageProcessingMixin from .image_processing_utils import ImageProcessingMixin
from .image_transforms import rescale, resize, to_pil_image
from .image_utils import ImageFeatureExtractionMixin from .image_utils import ImageFeatureExtractionMixin
from .models.beit import BeitFeatureExtractor, BeitImageProcessor from .models.beit import BeitFeatureExtractor, BeitImageProcessor
from .models.bit import BitImageProcessor from .models.bit import BitImageProcessor
......
...@@ -10,18 +10,6 @@ class ImageProcessingMixin(metaclass=DummyObject): ...@@ -10,18 +10,6 @@ class ImageProcessingMixin(metaclass=DummyObject):
requires_backends(self, ["vision"]) 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): class ImageFeatureExtractionMixin(metaclass=DummyObject):
_backends = ["vision"] _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