remove numba import