Try workaround for newer Numba

Author: pitrou

ci/scripts/python_test.sh

@@ -69,4 +69,4 @@ export PYARROW_TEST_PARQUET_ENCRYPTION
 export PYARROW_TEST_S3
 
 # Testing PyArrow
-pytest -r s ${PYTEST_ARGS} --pyargs pyarrow
+pytest -r s ${PYTEST_ARGS} --pyargs pyarrow.tests.test_cuda_numba_interop

python/pyarrow/_cuda.pyx

@@ -460,9 +460,14 @@ cdef class CudaBuffer(Buffer):
         """
         import ctypes
         from numba.cuda.cudadrv.driver import MemoryPointer
-        return MemoryPointer(self.context.to_numba(),
-                             pointer=ctypes.c_void_p(self.address),
-                             size=self.size)
+        try:
+            return MemoryPointer(self.context.to_numba(),
+                                 pointer=ctypes.c_void_p(self.address),
+                                 size=self.size)
+        except TypeError:
+            # Newer Numba does not take a context argument anymore
+            return MemoryPointer(pointer=ctypes.c_void_p(self.address),
+                                 size=self.size)
 
     cdef getitem(self, int64_t i):
         return self.copy_to_host(position=i, nbytes=1)[0]