update numba (#149)

* fix: get_thread_id api (closes #140)

numba/numba#8383

* release numba 0.56 lock

Author: RemyLau

requirements.txt

@@ -1,5 +1,5 @@
 gensim==4.2.0
-numpy==1.22.4
-numba==0.55.2
+numpy==1.23.2
+numba==0.56.2
 numba-progress==0.0.3
 nptyping==2.2.0

setup.cfg

@@ -30,7 +30,7 @@ keywords =
 install_requires =
     gensim>=4.1.0
     numpy>=1.20.0
-    numba>=0.46.0,<0.56.0
+    numba>=0.46.0
     numba-progress>=0.0.2
     nptyping>=2.0.0
 

src/pecanpy/pecanpy.py

@@ -3,7 +3,6 @@
 from gensim.models import Word2Vec
 from numba import njit
 from numba import prange
-from numba.np.ufunc.parallel import _get_thread_id
 from numba_progress import ProgressBar
 
 from .graph import BaseGraph
@@ -19,6 +18,11 @@
 from .typing import Uint64Array
 from .wrappers import Timer
 
+try:
+    from numba.np.ufunc.parallel import get_thread_id
+except ImportError:  # numba<0.56
+    from numba.np.ufunc.parallel import _get_thread_id as get_thread_id
+
 
 class Base(BaseGraph):
     """Base node2vec object.
@@ -173,7 +177,7 @@ def _random_walks(
         """Simulate a random walk starting from start node."""
         # Seed the random number generator
         if random_state is not None:
-            np.random.seed(random_state + _get_thread_id())
+            np.random.seed(random_state + get_thread_id())
 
         # use the last entry of each walk index array to keep track of the
         # effective walk length