make prototype module and therefore torchdata dependency optional

neuromorphs · Jan 12, 2023 · 367b27f · 367b27f
1 parent f4cb6fc
commit 367b27f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tonic/__init__.py b/tonic/__init__.py
@@ -1,9 +1,14 @@
 from pbr.version import VersionInfo
 
-from . import collation, datasets, io, prototype, slicers, transforms, utils
+from . import collation, datasets, io, slicers, transforms, utils
 from .cached_dataset import CachedDataset, DiskCachedDataset, MemoryCachedDataset
 from .dataset import Dataset
 from .sliced_dataset import SlicedDataset
 
+try:
+    from . import prototype
+except (ImportError, NameError):
+    pass
+
 all = "__version__"
 __version__ = VersionInfo("tonic").release_string()