ensure optional dependencies aren't imported

zarr-developers · Oct 19, 2024 · f6c75da · f6c75da
1 parent 55d152d
commit f6c75da
Show file tree

Hide file tree

Showing 6 changed files with 27 additions and 5 deletions.
diff --git a/virtualizarr/backend.py b/virtualizarr/backend.py
@@ -10,8 +10,16 @@
 from xarray import Dataset
 from xarray.core.indexes import Index
 
-from virtualizarr.readers import *
 from virtualizarr.manifests import ManifestArray
+from virtualizarr.readers import (
+    DMRPPVirtualBackend,
+    FITSVirtualBackend,
+    HDF5VirtualBackend,
+    KerchunkVirtualBackend,
+    NetCDF3VirtualBackend,
+    TIFFVirtualBackend,
+    ZarrV3VirtualBackend,
+)
 from virtualizarr.utils import _FsspecFSFromFilepath, check_for_collisions
 
 # TODO add entrypoint to allow external libraries to add to this mapping

diff --git a/virtualizarr/readers/__init__.py b/virtualizarr/readers/__init__.py
@@ -5,3 +5,13 @@
 from virtualizarr.readers.netcdf3 import NetCDF3VirtualBackend
 from virtualizarr.readers.tiff import TIFFVirtualBackend
 from virtualizarr.readers.zarr_v3 import ZarrV3VirtualBackend
+
+__all__ = [
+    "DMRPPVirtualBackend",
+    "FITSVirtualBackend",
+    "HDF5VirtualBackend",
+    "KerchunkVirtualBackend",
+    "NetCDF3VirtualBackend",
+    "TIFFVirtualBackend",
+    "ZarrV3VirtualBackend",
+]
diff --git a/virtualizarr/readers/fits.py b/virtualizarr/readers/fits.py
@@ -1,6 +1,5 @@
 from typing import Iterable, Mapping, Optional
 
-from kerchunk.fits import process_file
 from xarray import Dataset
 from xarray.core.indexes import Index
 
@@ -27,6 +26,8 @@ def open_virtual_dataset(
         indexes: Mapping[str, Index] | None = None,
         reader_options: Optional[dict] = None,
     ) -> Dataset:
+        from kerchunk.fits import process_file
+
         # handle inconsistency in kerchunk, see GH issue https://github.com/zarr-developers/VirtualiZarr/issues/160
         refs = KerchunkStoreRefs({"refs": process_file(filepath, **reader_options)})
 

diff --git a/virtualizarr/readers/hdf5.py b/virtualizarr/readers/hdf5.py
@@ -1,6 +1,5 @@
 from typing import Iterable, Mapping, Optional
 
-from kerchunk.hdf import SingleHdf5ToZarr
 from xarray import Dataset
 from xarray.core.indexes import Index
 
@@ -27,6 +26,8 @@ def open_virtual_dataset(
         indexes: Mapping[str, Index] | None = None,
         reader_options: Optional[dict] = None,
     ) -> Dataset:
+        from kerchunk.hdf import SingleHdf5ToZarr
+
         drop_variables, loadable_variables = check_for_collisions(
             drop_variables,
             loadable_variables,

diff --git a/virtualizarr/readers/netcdf3.py b/virtualizarr/readers/netcdf3.py
@@ -1,6 +1,5 @@
 from typing import Iterable, Mapping, Optional
 
-from kerchunk.netCDF3 import NetCDF3ToZarr
 from xarray import Dataset
 from xarray.core.indexes import Index
 
@@ -27,6 +26,8 @@ def open_virtual_dataset(
         indexes: Mapping[str, Index] | None = None,
         reader_options: Optional[dict] = None,
     ) -> Dataset:
+        from kerchunk.netCDF3 import NetCDF3ToZarr
+
         drop_variables, loadable_variables = check_for_collisions(
             drop_variables,
             loadable_variables,

diff --git a/virtualizarr/readers/tiff.py b/virtualizarr/readers/tiff.py
@@ -1,7 +1,6 @@
 import warnings
 from typing import Iterable, Mapping, Optional
 
-from kerchunk.tiff import tiff_to_zarr
 from xarray import Dataset
 from xarray.core.indexes import Index
 
@@ -29,6 +28,8 @@ def open_virtual_dataset(
         indexes: Mapping[str, Index] | None = None,
         reader_options: Optional[dict] = None,
     ) -> Dataset:
+        from kerchunk.tiff import tiff_to_zarr
+
         drop_variables, loadable_variables = check_for_collisions(
             drop_variables=drop_variables, loadable_variables=loadable_variables
         )