Simplify transpose in xr.dot

Illviljan · Illviljan · commit 63e1a350431b · 2021-10-10T20:44:25.000+02:00
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):

Original file line number	Diff line number	Diff line change
`@@ -1529,7 +1529,7 @@ def dot(arrays, dims=None, *kwargs):`
`1529`	`1529`	`join=join,`
`1530`	`1530`	`dask="allowed",`
`1531`	`1531`	`)`
`1532`		`- return result.transpose(*[d for d in all_dims if d in result.dims])`
	`1532`	`+ return result.transpose(*all_dims, missing_dims="ignore")`
`1533`	`1533`
`1534`	`1534`
`1535`	`1535`	`def where(cond, x, y):`