I think it's standard to use `np.squeeze` in this case.