From 542a953a73c420f26b086f61d0dec62af33ea290 Mon Sep 17 00:00:00 2001 From: Matthew Iannucci Date: Tue, 22 Oct 2024 07:29:05 -0400 Subject: [PATCH] Fix import errors in icechunk writer --- virtualizarr/writers/icechunk.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/virtualizarr/writers/icechunk.py b/virtualizarr/writers/icechunk.py index 3984d0e6..231262d1 100644 --- a/virtualizarr/writers/icechunk.py +++ b/virtualizarr/writers/icechunk.py @@ -34,8 +34,13 @@ def dataset_to_icechunk(ds: Dataset, store: "IcechunkStore") -> None: ds: xr.Dataset store: IcechunkStore """ - from icechunk import IcechunkStore # type: ignore[import-not-found] - from zarr import Group # type: ignore[import-untyped] + try: + from icechunk import IcechunkStore # type: ignore[import-not-found] + from zarr import Group # type: ignore[import-untyped] + except ImportError: + raise ImportError( + "The 'icechunk' and 'zarr' version 3 libraries are required to use this function" + ) if not isinstance(store, IcechunkStore): raise TypeError(f"expected type IcechunkStore, but got type {type(store)}")