Handle support for older xarray

microsoft · Sep 29, 2023 · 50470b3 · 50470b3
1 parent 6d22e34
commit 50470b3
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/qcodes/dataset/exporters/export_to_xarray.py b/qcodes/dataset/exporters/export_to_xarray.py
@@ -93,10 +93,15 @@ def _load_to_xarray_dataarray_dict_no_metadata(
                 on_grid = index_prod == len(index)
                 if not on_grid:
                     assert isinstance(df.index, pd.MultiIndex)
-                    coords = xr.Coordinates.from_pandas_multiindex(
-                        df.index, "multi_index"
-                    )
-                    xrdarray = xr.DataArray(df[name], coords=coords)
+
+                    if hasattr(xr, "Coordinates"):
+                        coords = xr.Coordinates.from_pandas_multiindex(
+                            df.index, "multi_index"
+                        )
+                        xrdarray = xr.DataArray(df[name], coords=coords)
+                    else:
+                        # support xarray < 2023.8.0, can be removed when we drop support for that
+                        xrdarray = xr.DataArray(df[name], [("multi_index", df.index)])
                 else:
                     xrdarray = df.to_xarray().get(name, xr.DataArray())