Disable PCodec if dependencies are unavailable

zarr-developers · Nov 18, 2024 · 0a8d9e2 · 0a8d9e2
1 parent 145f57c
commit 0a8d9e2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 9 deletions.
diff --git a/numcodecs/__init__.py b/numcodecs/__init__.py
@@ -144,6 +144,7 @@
 
 register_codec(Fletcher32)
 
-from numcodecs.pcodec import PCodec
+with suppress(ImportError):
+    from numcodecs.pcodec import PCodec
 
-register_codec(PCodec)
+    register_codec(PCodec)
diff --git a/numcodecs/pcodec.py b/numcodecs/pcodec.py
@@ -3,10 +3,7 @@
 from numcodecs.abc import Codec
 from numcodecs.compat import ensure_contiguous_ndarray
 
-try:
-    from pcodec import ChunkConfig, ModeSpec, PagingSpec, standalone
-except ImportError:  # pragma: no cover
-    standalone = None
+from pcodec import ChunkConfig, ModeSpec, PagingSpec, standalone
 
 
 DEFAULT_MAX_PAGE_N = 262144
@@ -49,9 +46,6 @@ def __init__(
         # TODO one day, add support for the Try* mode specs
         mode_spec: Literal['auto', 'classic'] = 'auto',
     ):
-        if standalone is None:  # pragma: no cover
-            raise ImportError("pcodec must be installed to use the PCodec codec.")
-
         # note that we use `level` instead of `compression_level` to
         # match other codecs
         self.level = level