Merge pull request #323 from mdekstrand/tweak/numba-0.56

Support Numba 0.56 (#319)
lenskit · Aug 16, 2022 · c9d1bf4 · c9d1bf4
2 parents ca4fba0 + 9a85e67
commit c9d1bf4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/lenskit/math/solve.py b/lenskit/math/solve.py
@@ -42,8 +42,10 @@ def _dposv(A, b, lower):
     # invert the meaning of 'lower' and 'trans', and the function will work fine.
     # We also need to swap index orders
     uplo = __uplo_U if lower else __uplo_L
-    n_p = __ffi.from_buffer(np.array([A.shape[0]], dtype=np.intc))
-    nrhs_p = __ffi.from_buffer(np.ones(1, dtype=np.intc))
+    narr = np.array([A.shape[0]], dtype=np.intc)
+    n_p = __ffi.from_buffer(narr)
+    nrhs = np.ones(1, dtype=np.intc)
+    nrhs_p = __ffi.from_buffer(nrhs)
     info = np.zeros(1, dtype=np.intc)
     info_p = __ffi.from_buffer(info)
 
@@ -52,7 +54,7 @@ def _dposv(A, b, lower):
             __ffi.from_buffer(b), n_p,
             info_p)
 
-    _ref_sink(n_p, nrhs_p, info, info_p)
+    _ref_sink(narr, n_p, nrhs, nrhs_p, info, info_p)
 
     return info[0]
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -24,12 +24,12 @@ dependencies = [
     "pandas >=1.0, ==1.*",
     "numpy >= 1.19",
     "scipy >= 1.2",
-    "numba >= 0.51, < 0.56",
+    "numba >= 0.51, < 0.57",
     "cffi >= 1.12.2",
     "psutil >= 5",
     "binpickle >= 0.3.2",
     "seedbank >= 0.1.0",
-    "csr >= 0.3.1",
+    "csr >= 0.4",
 ]
 
 [project.urls]