directly returning a xr dataset when loading from netcdf file

microsoft · Jan 16, 2024 · 772dad1 · 772dad1
1 parent aa69eda
commit 772dad1
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/qcodes/dataset/data_set_cache.py b/src/qcodes/dataset/data_set_cache.py
@@ -465,6 +465,18 @@ def load_data_from_db(self) -> None:
             )
 
 
+    def to_xarray_dataset(
+        self, *, use_multi_index: Literal["auto", "always", "never"] = "auto"
+    ) -> xr.Dataset:
+        if use_multi_index == "always":
+            ds = self._xr_dataset.stack()
+        elif use_multi_index == "never":
+            ds = self._xr_dataset.unstack()
+        else:
+            ds = self._xr_dataset
+        return ds
+
+
 class DataSetCacheWithDBBackend(DataSetCache["DataSet"]):
     def load_data_from_db(self) -> None:
         """