Simplify transpose in xr.dot

pydata · Oct 10, 2021 · 63e1a35 · 63e1a35
1 parent c98579f
commit 63e1a35
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xarray/core/computation.py b/xarray/core/computation.py
@@ -1529,7 +1529,7 @@ def dot(*arrays, dims=None, **kwargs):
         join=join,
         dask="allowed",
     )
-    return result.transpose(*[d for d in all_dims if d in result.dims])
+    return result.transpose(*all_dims, missing_dims="ignore")
 
 
 def where(cond, x, y):