-
Notifications
You must be signed in to change notification settings - Fork 25
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add Icechunk Support #256
Add Icechunk Support #256
Conversation
for more information, see https://pre-commit.ci
Okay thinnkkkkk were finally solid |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I'm psyched about this! Just some last few commments.
# Check with xarray | ||
ds = open_zarr(store=icechunk_filestore, zarr_format=3, consolidated=False) | ||
assert np.allclose(ds.air.to_numpy(), expected_ds.air.to_numpy()) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
assert_identical
here too
root_group = group(store=icechunk_filestore) | ||
air_array = root_group["air"] | ||
assert isinstance(air_array, Array) | ||
assert air_array.shape == (3, 4) | ||
assert air_array.dtype == np.dtype("float64") | ||
assert air_array.attrs["units"] == "km" | ||
assert np.allclose(air_array[:], la_v[:]) | ||
|
||
pres_array = root_group["pres"] | ||
assert isinstance(pres_array, Array) | ||
assert pres_array.shape == (3, 4) | ||
assert pres_array.dtype == np.dtype("int32") | ||
expected_ds = open_dataset(simple_netcdf4) | ||
expected_array = expected_ds["foo"].to_numpy() | ||
npt.assert_equal(pres_array, expected_array) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This can also all be checked by doing open_zarr
then using a single xarray assertion. But it might be valuable to keep these lower-level assertions too.
Also add a note to |
Co-authored-by: Tom Nicholas <[email protected]>
There is some weirdness when comparing the icechunk datasets using the testing tools that dont happen if you check the individual peices of the dataset. Leaving further refinement for another PR |
So psyched for this!!! |
Adds the ability to write to an Icechunk store. Co developed with @TomNicholas.
See earth-mover/VirtualiZarr#1 for more information (ported from that branch)
docs/releases.rst
api.rst