Fix type checking consistency

jenshnielsen · jenshnielsen · commit 49073aa58bcb · 2025-09-03T10:43:10.000+02:00
diff --git a/src/qcodes/dataset/data_set.py b/src/qcodes/dataset/data_set.py
@@ -1020,7 +1020,7 @@ def to_xarray_dataarray_dict(
             self, data, use_multi_index=use_multi_index
         )
 
-        return datadict
+        return {key: val[key] for key, val in datadict.items()}
 
     def to_xarray_dataset(
         self,
diff --git a/src/qcodes/dataset/data_set_cache.py b/src/qcodes/dataset/data_set_cache.py
@@ -200,9 +200,10 @@ def to_xarray_dataarray_dict(
 
         """
         data = self.data()
-        return load_to_xarray_dataarray_dict(
+        data_dict = load_to_xarray_dataarray_dict(
             self._dataset, data, use_multi_index=use_multi_index
         )
+        return {key: val[key] for key, val in data_dict.items()}
 
     def to_xarray_dataset(
         self, *, use_multi_index: Literal["auto", "always", "never"] = "auto"

Original file line number	Diff line number	Diff line change
`@@ -1020,7 +1020,7 @@ def to_xarray_dataarray_dict(`
`1020`	`1020`	`self, data, use_multi_index=use_multi_index`
`1021`	`1021`	`)`
`1022`	`1022`
`1023`		`- return datadict`
	`1023`	`+ return {key: val[key] for key, val in datadict.items()}`
`1024`	`1024`
`1025`	`1025`	`def to_xarray_dataset(`
`1026`	`1026`	`self,`