wrap Dataset input in a DataTree node before setting

pydata · Sep 18, 2024 · b520f9c · b520f9c
1 parent bf036ba
commit b520f9c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/xarray/core/datatree.py b/xarray/core/datatree.py
@@ -895,6 +895,8 @@ def __setitem__(
             # TODO should possibly deal with hashables in general?
             # path-like: a name of a node/variable, or path to a node/variable
             path = NodePath(key)
+            if isinstance(value, Dataset):
+                value = DataTree(dataset=value)
             return self._set_item(path, value, new_nodes_along_path=True)
         else:
             raise ValueError("Invalid format for key")