Merge pull request #1759 from zm711/tostring

KwikIO: Limit IO to numpy < 2.3.0

Author: alejoe91

neo/io/kwikio.py

@@ -20,8 +20,12 @@
 from neo.io.baseio import BaseIO
 
 # to import from core
-from neo.core import Segment, SpikeTrain, AnalogSignal, Block, Group
+from neo.core import Segment, SpikeTrain, AnalogSignal, Block, Group, NeoReadWriteError
 
+import importlib.metadata
+from packaging.version import Version, parse
+
+numpy_version = parse(importlib.metadata.version('numpy'))
 
 class KwikIO(BaseIO):
     """
@@ -58,6 +62,12 @@ def __init__(self, filename):
         Arguments:
             filename : the filename
         """
+
+        if numpy_version >= Version("2.3.0"):
+            deprecation_msg = ("As the kwik package is no longer actively maintained it only works with NumPy < 2.3.0 and your environment "
+            f"has {numpy_version}. Downgrade your NumPy version to use this IO")
+            raise NeoReadWriteError(deprecation_msg)
+        
         from klusta import kwik
 
         BaseIO.__init__(self)

neo/test/iotest/test_kwikio.py

@@ -3,12 +3,17 @@
 """
 
 import unittest
-
-try:
-    from klusta import kwik
-
+import importlib.util
+import importlib.metadata
+from packaging.version import Version, parse
+
+kwik_spec = importlib.util.find_spec('klusta')
+# kwik no longer works with recent versions of numpy
+numpy_version = parse(importlib.metadata.version('numpy'))
+numpy_okay = numpy_version < Version('2.3.0')
+if kwik_spec is not None and numpy_okay:
     HAVE_KWIK = True
-except ImportError:
+else:
     HAVE_KWIK = False
 
 from neo.io import kwikio