Update dataset_plot.py

pydata · Jul 4, 2024 · f2d7145 · f2d7145
1 parent e6922d5
commit f2d7145
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/xarray/plot/dataset_plot.py b/xarray/plot/dataset_plot.py
@@ -732,9 +732,10 @@ def _temp_dataarray(ds: Dataset, y: Hashable, locals_: dict[str, Any]) -> DataAr
     coord_kwargs = locals_.keys() & valid_coord_kwargs
     for k in coord_kwargs:
         key = locals_[k]
-        if key is not None:
-            coords[key] = ds[key]
-            for d in ds[key].dims:
+        darray = ds.get(key)
+        if darray is not None:
+            coords[key] = darray
+            for d in darray.dims:
                 dims[d] = None
 
     # Trim dataset from unneccessary dims: