removing numba import in grpah