uxarray 2026.3.0__tar.gz → 2026.4.0__tar.gz

{uxarray-2026.3.0 → uxarray-2026.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uxarray
-Version: 2026.3.0
+Version: 2026.4.0
 Summary: Xarray extension for unstructured climate and global weather data analysis and visualization.
 License:                                  Apache License
                                    Version 2.0, January 2004

{uxarray-2026.3.0 → uxarray-2026.4.0}/uxarray/core/dataset.py

@@ -90,6 +90,24 @@ class UxDataset(xr.Dataset):
         else:
             self._uxgrid = uxgrid
 
+        # As of xarray's 2026.4.0, `xr.Dataset(xr.Dataset)` is prohibited;
+        # hence this check, i.e. if we get `xr.Dataset` as input, use its `data_vars`
+        # as `dict` and handle `coords` and `attrs` properly as well
+        if args and isinstance(args[0], xr.Dataset):
+            ds = args[0]
+            # Replacee only args[0], `ds`, with `ds.data_vars` as `dict`
+            args = (dict(ds.data_vars),) + args[1:]
+            # coords not passed positionally
+            if len(args) < 2:
+                kwargs.setdefault(
+                    "coords", dict(ds.coords)
+                )  # Set it as kwarg only if not explicitly provided
+            # attrs not passed positionally
+            if len(args) < 3:
+                kwargs.setdefault(
+                    "attrs", ds.attrs
+                )  # Set it as kwarg only if not explicitly provided
+
         super().__init__(*args, **kwargs)
 
     # declare plotting accessor
@@ -627,9 +645,9 @@ class UxDataset(xr.Dataset):
         """
         if grid_format == "HEALPix":
             ds = self.rename_dims({"n_face": "cell"})
-            return xr.Dataset(ds)
+            return xr.Dataset(ds.data_vars, coords=ds.coords, attrs=ds.attrs)
 
-        return xr.Dataset(self)
+        return xr.Dataset(self.data_vars, coords=self.coords, attrs=self.attrs)
 
     def get_dual(self):
         """Compute the dual mesh for a dataset, returns a new dataset object.

{uxarray-2026.3.0 → uxarray-2026.4.0}/uxarray/io/utils.py

@@ -7,51 +7,114 @@ from uxarray.io._mpas import _mpas_to_ugrid_dims
 from uxarray.io._ugrid import _is_ugrid, _read_ugrid
 
 
+def _is_exodus(dataset: xr.Dataset) -> bool:
+    """Check whether a dataset looks like an Exodus mesh."""
+    has_packed_coords = "coord" in dataset
+    has_split_coords = {"coordx", "coordy"}.issubset(dataset.variables)
+    has_connectivity = any(
+        name.startswith("connect") for name in dataset.variables
+    ) or any("num_nod_per_el" in dim for dim in dataset.dims)
+
+    return has_connectivity and (has_packed_coords or has_split_coords)
+
+
+def _is_scrip(dataset: xr.Dataset) -> bool:
+    """Check whether a dataset looks like an unstructured SCRIP grid."""
+    required_vars = {
+        "grid_center_lon",
+        "grid_center_lat",
+        "grid_corner_lon",
+        "grid_corner_lat",
+    }
+    unstructured_markers = {"grid_imask", "grid_rank", "grid_area"}
+
+    return required_vars.issubset(dataset.variables) and any(
+        marker in dataset for marker in unstructured_markers
+    )
+
+
+def _is_mpas(dataset: xr.Dataset) -> bool:
+    """Check whether a dataset looks like an MPAS grid."""
+    if "verticesOnCell" not in dataset:
+        return False
+
+    companion_groups = (
+        {"nEdgesOnCell"},
+        {"latCell", "lonCell"},
+        {"latVertex", "lonVertex"},
+        {"xCell", "yCell", "zCell"},
+        {"xVertex", "yVertex", "zVertex"},
+    )
+
+    return any(group.issubset(dataset.variables) for group in companion_groups)
+
+
+def _is_esmf(dataset: xr.Dataset) -> bool:
+    """Check whether a dataset looks like an ESMF mesh."""
+    return "maxNodePElement" in dataset.dims and "elementConn" in dataset
+
+
+def _is_geos_cs(dataset: xr.Dataset) -> bool:
+    """Check whether a dataset looks like a GEOS cube-sphere grid."""
+    required_dims = {"nf", "YCdim", "XCdim"}
+    required_vars = {"corner_lons", "corner_lats"}
+
+    return required_dims.issubset(dataset.sizes) and required_vars.issubset(
+        dataset.variables
+    )
+
+
+def _is_icon(dataset: xr.Dataset) -> bool:
+    """Check whether a dataset looks like an ICON grid."""
+    required_vars = {"vertex_of_cell", "clon", "clat", "vlon", "vlat"}
+    return required_vars.issubset(dataset.variables)
+
+
+def _is_fesom2(dataset: xr.Dataset) -> bool:
+    """Check whether a dataset looks like a FESOM2 grid."""
+    return "triag_nodes" in dataset
+
+
 def _parse_grid_type(dataset):
-    """Checks input and contents to determine grid type. Supports detection of
-    UGrid, SCRIP, Exodus, ESMF, and shape file.
+    """Determine the grid type represented by an input dataset.
 
     Parameters
     ----------
     dataset : Xarray dataset
-       Xarray dataset of the grid
+        Xarray dataset containing grid topology information.
 
     Returns
     -------
-    mesh_type : str
-        File type of the file, ug, exo, scrip or shp
+    tuple[str, str | None, str | None]
+        A 3-tuple of ``(mesh_type, lon_name, lat_name)``. ``mesh_type`` is one
+        of ``"Exodus"``, ``"Scrip"``, ``"UGRID"``, ``"MPAS"``, ``"ESMF"``,
+        ``"GEOS-CS"``, ``"ICON"``, ``"FESOM2"``, or ``"Structured"``. The
+        longitude and latitude coordinate names are only returned for structured
+        grids and are otherwise ``None``.
 
     Raises
     ------
     RuntimeError
-            If invalid file type
-    ValueError
-        If file is not in UGRID format
+        If the dataset format cannot be recognized.
     """
 
     _structured, lon_name, lat_name = _is_structured(dataset)
 
-    if "coord" in dataset:
-        # exodus with coord or coordx
-        mesh_type = "Exodus"
-    elif "coordx" in dataset:
+    if _is_exodus(dataset):
         mesh_type = "Exodus"
-    elif "grid_center_lon" in dataset:
-        # scrip with grid_center_lon
+    elif _is_scrip(dataset):
         mesh_type = "Scrip"
     elif _is_ugrid(dataset):
-        # ugrid topology is present
         mesh_type = "UGRID"
-    elif "verticesOnCell" in dataset:
+    elif _is_mpas(dataset):
         mesh_type = "MPAS"
-    elif "maxNodePElement" in dataset.dims:
+    elif _is_esmf(dataset):
         mesh_type = "ESMF"
-    elif all(key in dataset.sizes for key in ["nf", "YCdim", "XCdim"]):
-        # expected dimensions for a GEOS cube sphere grid
+    elif _is_geos_cs(dataset):
         mesh_type = "GEOS-CS"
-    elif "vertex_of_cell" in dataset:
+    elif _is_icon(dataset):
         mesh_type = "ICON"
-    elif "triag_nodes" in dataset:
+    elif _is_fesom2(dataset):
         mesh_type = "FESOM2"
     elif _structured:
         mesh_type = "Structured"

{uxarray-2026.3.0 → uxarray-2026.4.0}/uxarray.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uxarray
-Version: 2026.3.0
+Version: 2026.4.0
 Summary: Xarray extension for unstructured climate and global weather data analysis and visualization.
 License:                                  Apache License
                                    Version 2.0, January 2004