Fixes

pytorch · Aug 18, 2023 · 7f8ed88 · 7f8ed88
1 parent d3782e2
commit 7f8ed88
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/torchvision/datasets/__init__.py b/torchvision/datasets/__init__.py
@@ -1,5 +1,3 @@
-from torchvision.datapoints._dataset_wrapper import wrap_dataset_for_transforms_v2
-
 from ._optical_flow import FlyingChairs, FlyingThings3D, HD1K, KittiFlow, Sintel
 from ._stereo_matching import (
     CarlaStereo,
@@ -131,3 +129,16 @@
     "ETH3DStereo",
     "wrap_dataset_for_transforms_v2",
 )
+
+
+# We override current module's attributes to handle the import:
+# from torchvision.datasets import wrap_dataset_for_transforms_v2
+# without a cyclic error.
+# Ref: https://peps.python.org/pep-0562/
+def __getattr__(name):
+    if name in ("wrap_dataset_for_transforms_v2",):
+        from torchvision.datapoints._dataset_wrapper import wrap_dataset_for_transforms_v2
+
+        return wrap_dataset_for_transforms_v2
+
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")