perf(datasets): lazily load datasets in init files (video)

Signed-off-by: Deepyaman Datta <[email protected]>
kedro-org · Jul 29, 2023 · 51f2f46 · 51f2f46
1 parent c95be7b
commit 51f2f46
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/kedro-datasets/kedro_datasets/video/__init__.py b/kedro-datasets/kedro_datasets/video/__init__.py
@@ -1,5 +1,11 @@
 """Dataset implementation to load/save data from/to a video file."""
+from typing import Any
 
-__all__ = ["VideoDataSet"]
+import lazy_loader as lazy
 
-from .video_dataset import VideoDataSet
+# https://github.com/pylint-dev/pylint/issues/4300#issuecomment-1043601901
+VideoDataSet: Any
+
+__getattr__, __dir__, __all__ = lazy.attach(
+    __name__, submod_attrs={"video_dataset": ["VideoDataSet"]}
+)