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

You're sure? I always lean on the side of raising errors rather than
accepting input from the user which might be false.