Better chooce.infer_shape error.