cleaning up some numba imports