Skip to content

Commit

Permalink
Better gate imports
Browse files Browse the repository at this point in the history
  • Loading branch information
jrbourbeau committed Aug 15, 2023
1 parent a303705 commit 61d256b
Showing 1 changed file with 9 additions and 10 deletions.
19 changes: 9 additions & 10 deletions earthaccess/kerchunk.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,15 +6,6 @@
import fsspec
import s3fs

try:
from kerchunk.combine import MultiZarrToZarr
from kerchunk.hdf import SingleHdf5ToZarr
except ImportError as e:
raise ImportError(
"`earthaccess.consolidate_metadata` requires `kerchunk` to be be installed"
) from e


try:
from dask import compute, delayed
except ImportError:
Expand All @@ -33,10 +24,10 @@ def get_chunk_metadata(
granuale: earthaccess.results.DataGranule,
fs: fsspec.AbstractFileSystem | s3fs.S3FileSystem,
) -> list[dict]:
from kerchunk.hdf import SingleHdf5ToZarr
metadata = []
access = "direct" if isinstance(fs, s3fs.S3FileSystem) else "indirect"
for url in granuale.data_links(access=access):
print(f"{url = }")
with fs.open(url) as inf:
h5chunks = SingleHdf5ToZarr(inf, url)
m = h5chunks.translate()
Expand All @@ -51,6 +42,14 @@ def consolidate_metadata(
kerchunk_options: dict | None = None,
access: str = "direct",
) -> str:

try:
from kerchunk.combine import MultiZarrToZarr
except ImportError as e:
raise ImportError(
"`earthaccess.consolidate_metadata` requires `kerchunk` to be be installed"
) from e

if access == "direct":
fs = earthaccess.get_s3fs_session(provider=granuales[0]["meta"]["provider-id"])
else:
Expand Down

0 comments on commit 61d256b

Please sign in to comment.