Check for DataArray instance (pydata#7030)

dranjan · Aug 12, 2023 · b22eada · b22eada
1 parent 21c400c
commit b22eada
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/xarray/core/variable.py b/xarray/core/variable.py
@@ -943,8 +943,13 @@ def __setitem__(self, key, value):
 
         See __getitem__ for more details.
         """
+        from xarray.core.dataarray import DataArray
+
         dims, index_tuple, new_order = self._broadcast_indexes(key)
 
+        if isinstance(value, DataArray):
+            value = value.variable
+
         if not isinstance(value, Variable):
             value = as_compatible_data(value)
             if value.ndim > len(dims):