Python 3.8 compatibility

geometric-kernels · Oct 24, 2024 · b84b03f · b84b03f
1 parent 50caf1b
commit b84b03f
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/geometric_kernels/lab_extras/numpy/sparse_extras.py b/geometric_kernels/lab_extras/numpy/sparse_extras.py
@@ -1,3 +1,5 @@
+import sys
+
 import lab as B
 import scipy
 import scipy.sparse as sp
@@ -11,10 +13,13 @@
 SparseArray defines a lab data type that covers all possible sparse
 scipy arrays, so that multiple dispatch works with such arrays.
 """
-SparseArray = Union[
-    sp.sparray,
-    sp.spmatrix,
-]
+if sys.version_info[:2] <= (3, 9):
+    SparseArray = sp.spmatrix
+else:
+    SparseArray = Union[
+        sp.sparray,
+        sp.spmatrix,
+    ]
 
 
 @dispatch