subset.bounding_circle error during plotting

[mathomp4]

trafficstars

Version

v2024.06.0

How did you install UXarray?

Pip

What happened?

With the support for GEOS CS files via #802 in v2024.06.0, I decided to try it out and duplicate the plots by @philipc2 as seen here.

I was able to do the global plot, but the regional plot failed with:

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
Cell In[6], line 1
----> 1 uxds['RH'][0][0].subset.bounding_circle((-90, 25), 10).plot.rasterize(method='polygon', exclude_antimeridian=True, pixel_ratio=4.0) * features

File ~/installed/Core/GEOSpyD/24.4.0-0_py3.11/2024-06-11/lib/python3.11/site-packages/uxarray/subset/dataarray_accessor.py:85, in DataArraySubsetAccessor.bounding_circle(self, center_coord, r, element, **kwargs)
     70 """Subsets an unstructured grid by returning all elements within some
     71 radius (in degrees) from a center coord.
     72 
   (...)
     80     Element for use with `coords` comparison, one of `nodes`, `face centers`, or `edge centers`
     81 """
     82 grid = self.uxda.uxgrid.subset.bounding_circle(
     83     center_coord, r, element, **kwargs
     84 )
---> 85 return self.uxda._slice_from_grid(grid)

File ~/installed/Core/GEOSpyD/24.4.0-0_py3.11/2024-06-11/lib/python3.11/site-packages/uxarray/core/dataarray.py:1092, in UxDataArray._slice_from_grid(self, sliced_grid)
   1087 else:
   1088     raise ValueError(
   1089         "Data variable must be either node, edge, or face centered."
   1090     )
-> 1092 return UxDataArray(
   1093     uxgrid=sliced_grid,
   1094     data=d_var,
   1095     name=self.name,
   1096     coords=self.coords,
   1097     dims=self.dims,
   1098     attrs=self.attrs,
   1099 )

File ~/installed/Core/GEOSpyD/24.4.0-0_py3.11/2024-06-11/lib/python3.11/site-packages/uxarray/core/dataarray.py:71, in UxDataArray.__init__(self, uxgrid, *args, **kwargs)
     68 else:
     69     self.uxgrid = uxgrid
---> 71 super().__init__(*args, **kwargs)

File ~/installed/Core/GEOSpyD/24.4.0-0_py3.11/2024-06-11/lib/python3.11/site-packages/xarray/core/dataarray.py:455, in DataArray.__init__(self, data, coords, dims, name, attrs, indexes, fastpath)
    453 data = _check_data_shape(data, coords, dims)
    454 data = as_compatible_data(data)
--> 455 coords, dims = _infer_coords_and_dims(data.shape, coords, dims)
    456 variable = Variable(dims, data, attrs, fastpath=True)
    458 if not isinstance(coords, Coordinates):

File ~/installed/Core/GEOSpyD/24.4.0-0_py3.11/2024-06-11/lib/python3.11/site-packages/xarray/core/dataarray.py:194, in _infer_coords_and_dims(shape, coords, dims)
    191             var.dims = (dim,)
    192             new_coords[dim] = var.to_index_variable()
--> 194 _check_coords_dims(shape, new_coords, dims_tuple)
    196 return new_coords, dims_tuple

File ~/installed/Core/GEOSpyD/24.4.0-0_py3.11/2024-06-11/lib/python3.11/site-packages/xarray/core/dataarray.py:128, in _check_coords_dims(shape, coords, dim)
    126 for d, s in v.sizes.items():
    127     if s != sizes[d]:
--> 128         raise ValueError(
    129             f"conflicting sizes for dimension {d!r}: "
    130             f"length {sizes[d]} on the data but length {s} on "
    131             f"coordinate {k!r}"
    132         )

ValueError: conflicting sizes for dimension 'n_face': length 18392 on the data but length 437400 on coordinate 'lons'

As I'm using the same code and the same file, my guess is maybe I'm missing some package? (I'm still a newbie at uxarray).

What did you expect to happen?

I was expecting to duplicate the image:

Can you provide a MCVE to repoduce the bug?

import uxarray as ux
import geoviews.feature as gf
import cartopy.crs as ccrs

features = gf.coastline(
    projection=ccrs.PlateCarree(), line_width=1, scale="50m"
) * gf.states(projection=ccrs.PlateCarree(), line_width=1, scale="50m")

file_path = r"test-fixapp-c270.geosgcm_prog_nat.20200415_0000z.nc4"
uxds = ux.open_dataset(file_path, file_path)
uxds['RH'][0][0].subset.bounding_circle((-90, 25), 10).plot.rasterize(method='polygon', exclude_antimeridian=True, pixel_ratio=4.0) * features

[philipc2]

Do you mind sharing what uxds['RH'] looks like?

EDIT:

Forgot I had this dataset, I'll look into it this afternoon and see if I can replicate the issue

[erogluorhan]

Hi @philipc2 are you still planning to look into this or should we consider someone else?

[philipc2]

Hi @philipc2 are you still planning to look into this or should we consider someone else?

Thanks for the reminder! I will look at this in the morning

[philipc2]

Hi @mathomp4

Can you try installing the latest version of UXarray and try running the following?

uxds['RH'][0][0].subset.bounding_circle((-90, 25), 10).plot(coastline='50m', backend='matplotlib')

This is the plot I was able to generate. Let me know if you need any help!

[mathomp4]

@philipc2 Ah. Okay. I had uxarray 2024.10.0 and saw the failure. After a pip upgrade to 2024.10.1, this seems to be fixed:

Not sure what in that patch fixed it, but it seems fixed.