Fix dtype issue.

dgbowl · Feb 22, 2024 · dd449c7 · dd449c7
1 parent b472f52
commit dd449c7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -42,7 +42,7 @@
         "chemicals>=1.0.0",
         "rdkit>=2022.3",
         "h5netcdf~=1.0",
-        "xarray < 2023.11",
+        "xarray ~= 2024.02",
         "xarray-datatree>=0.0.12",
         "dgbowl-schemas>=116",
         "matplotlib>=3.5.0",

diff --git a/src/dgpost/utils/extract.py b/src/dgpost/utils/extract.py
@@ -256,7 +256,7 @@ def get_key_recurse(dt, keys):
         if len(vals.coords.dims) > 1:
             assert "uts" in vals.coords.dims
             for coord in vals.coords.dims:
-                if vals.coords.dtypes[coord] in {np.dtype("O")}:
+                if vals.coords.dtypes[coord].kind in {"O", "U"}:
                     splits = vals[coord].values
                     break
         else: