Add numba-level import for module type inference code