Update Numba version to match np.broadcast_to support

.github/workflows/test.yml

@@ -142,7 +142,7 @@ jobs:
         shell: bash -l {0}
         run: |
           mamba install --yes -q "python~=${PYTHON_VERSION}=*_cpython" mkl numpy scipy pip mkl-service graphviz cython libgpuarray pygpu
-          if [[ $INSTALL_NUMBA == "1" ]]; then mamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}=*_cpython" "numba>=0.52" numba-scipy; fi
+          if [[ $INSTALL_NUMBA == "1" ]]; then mamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}=*_cpython" "numba>=0.55" numba-scipy; fi
           mamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}=*_cpython" jax jaxlib
           pip install -q -r requirements.txt
           mamba list && pip freeze

environment.yml

@@ -21,7 +21,7 @@ dependencies:
   - mkl-service
   - libblas=*=*mkl
   # numba backend
-  - numba
+  - numba>=0.55
   - numba-scipy
   # GPU
   - libgpuarray

requirements.txt

@@ -13,7 +13,7 @@ sympy
 versioneer
 jax
 jaxlib
-numba
+numba>=0.55
 numba-scipy>=0.3.0
 diff-cover
 pre-commit