Fix ZFPY import error

docs/release.rst

@@ -31,6 +31,8 @@ Fix
   By :user:`Elliott Sales de Andrade <QuLogic>`, :issue:`487`.
 * Fix Upgrade to Zstd 1.5.5 due to potential corruption.
   By :user:`Mark Kittisopikul <mkitti>`, :issue:`429`
+* Add version constraint(<2.0) for numpy in zfpy.
+  By :user:`Tom Liang <px39n>`, :issue:`540`.
 
 Maintenance
 ~~~~~~~~~~~

numcodecs/zfpy.py

@@ -1,9 +1,26 @@
 from contextlib import suppress
+from importlib.metadata import PackageNotFoundError, version
+import warnings
 
 _zfpy = None
-with suppress(ImportError):
-    import zfpy as _zfpy
 
+_zfpy_version: tuple = ()
+with suppress(PackageNotFoundError):
+    _zfpy_version = tuple(map(int, version("zfpy").split(".")))
+
+if _zfpy_version:
+    # Check NumPy version
+    _numpy_version: tuple = tuple(map(int, version("numpy").split('.')))
+    if _numpy_version >= (2, 0, 0) and _zfpy_version <= (1, 0, 1):
+        _zfpy_version = ()
+        warnings.warn(
+            "NumPy version >= 2.0.0 detected. The zfpy library is incompatible with this version of NumPy. "
+            "Please downgrade to NumPy < 2.0.0 or wait for an update from zfpy.",
+            UserWarning,
+        )
+    else:
+        with suppress(ImportError):
+            import zfpy as _zfpy
 
 if _zfpy:
     from .abc import Codec

pyproject.toml

@@ -64,6 +64,7 @@ msgpack = [
 ]
 zfpy = [
     "zfpy>=1.0.0",
+    "numpy<2.0.0",    
 ]
 pcodec = [
     "pcodec>=0.1.0",