Is your feature request related to a problem? Please describe.
Zarr 3.x moved a number of types around and few other compatibility breaking changes. We currently try to support 2.x and 3.x in places like here but we should consider when to drop support for 2.x. This would help improve the typing of this module as well.