invprb[nan_mask] = np.nan