Use __all__ in lindi/__init__.py

NeurodataWithoutBorders · Sep 24, 2024 · 0533a66 · 0533a66
1 parent 759be92
commit 0533a66
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 5 deletions.
diff --git a/lindi/LindiH5pyFile/__init__.py b/lindi/LindiH5pyFile/__init__.py
@@ -2,11 +2,13 @@
 from .LindiH5pyDataset import LindiH5pyDataset
 from .LindiH5pyGroup import LindiH5pyGroup
 from .LindiH5pyLink import LindiH5pySoftLink, LindiH5pyHardLink
+from .LindiH5pyReference import LindiH5pyReference
 
 __all__ = [
     "LindiH5pyFile",
     "LindiH5pyDataset",
     "LindiH5pyGroup",
     "LindiH5pySoftLink",
     "LindiH5pyHardLink",
+    "LindiH5pyReference",
 ]
diff --git a/lindi/__init__.py b/lindi/__init__.py
@@ -1,5 +1,25 @@
-from .LindiH5ZarrStore import LindiH5ZarrStore, LindiH5ZarrStoreOpts  # noqa: F401
-from .LindiH5pyFile import LindiH5pyFile, LindiH5pyGroup, LindiH5pyDataset, LindiH5pyHardLink, LindiH5pySoftLink  # noqa: F401
-from .LocalCache.LocalCache import LocalCache, ChunkTooLargeError  # noqa: F401
-from .LindiRemfile.additional_url_resolvers import add_additional_url_resolver  # noqa: F401
-from .LindiH5pyFile.LindiH5pyReference import LindiH5pyReference  # noqa: F401
+from .LindiH5ZarrStore import LindiH5ZarrStore, LindiH5ZarrStoreOpts
+from .LindiH5pyFile import (
+    LindiH5pyFile,
+    LindiH5pyGroup,
+    LindiH5pyDataset,
+    LindiH5pyHardLink,
+    LindiH5pySoftLink,
+    LindiH5pyReference,
+)
+from .LocalCache.LocalCache import LocalCache, ChunkTooLargeError
+from .LindiRemfile.additional_url_resolvers import add_additional_url_resolver
+
+__all__ = [
+    "LindiH5ZarrStore",
+    "LindiH5ZarrStoreOpts",
+    "LindiH5pyFile",
+    "LindiH5pyGroup",
+    "LindiH5pyDataset",
+    "LindiH5pyHardLink",
+    "LindiH5pySoftLink",
+    "LindiH5pyReference",
+    "LocalCache",
+    "ChunkTooLargeError",
+    "add_additional_url_resolver",
+]