Fix tensor_to_image

pooya-mohammadi · Jul 20, 2024 · 075ad95 · 075ad95
1 parent 9e91692
commit 075ad95
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 10 deletions.
diff --git a/deep_utils/__init__.py b/deep_utils/__init__.py
@@ -4,7 +4,7 @@
 from .utils.lib_utils.integeration_utils import import_lazy_module
 
 # Deep Utils version number
-__version__ = "1.3.33"
+__version__ = "1.3.34"
 
 from .utils.constants import DUMMY_PATH, Backends
 

diff --git a/deep_utils/augmentation/cutmix/__init__.py b/deep_utils/augmentation/cutmix/__init__.py
@@ -1,7 +1,7 @@
-from .cutmix_tf import CutMixTF
-
-try:
-    from deep_utils.augmentation.cutmix.cutmix_torch import CutMixTorch
-    from deep_utils.dummy_objects.augmentation.cutmix import CutMixTorch
-except:
-    pass
+# from .cutmix_tf import CutMixTF
+#
+# try:
+#     from deep_utils.augmentation.cutmix.cutmix_torch import CutMixTorch
+#     from deep_utils.dummy_objects.augmentation.cutmix import CutMixTorch
+# except:
+#     pass
diff --git a/deep_utils/augmentation/torch/__init__.py b/deep_utils/augmentation/torch/__init__.py
@@ -1 +1 @@
-from .augmentation_torch import AugmentTorch, tensor_to_image, visualize_data_loader
+# from .augmentation_torch import AugmentTorch, tensor_to_image, visualize_data_loader
diff --git a/setup.py b/setup.py
@@ -1,6 +1,6 @@
 import setuptools
 
-VERSION = "1.3.33"
+VERSION = "1.3.34"
 
 long_description = open("Readme.md", mode="r", encoding="utf-8").read()