add [...] to transpose to allow for dimensions not present

AI4S2S · Feb 12, 2024 · bb0189f · bb0189f
1 parent 1b0b2d4
commit bb0189f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/s2spy/preprocess.py b/s2spy/preprocess.py
@@ -79,7 +79,7 @@ def _get_lineartrend_timeseries(data: Union[xr.DataArray, xr.Dataset], trend: di
     trend_timeseries = trend["intercept"] + trend["slope"] * (
         data["time"].astype(float)
     )
-    return trend_timeseries.transpose(*data.dims)
+    return trend_timeseries.transpose(*list(data.dims) + [...])
 
 
 def _trend_poly(data: Union[xr.DataArray, xr.Dataset], degree: int = 2) -> dict:
@@ -140,7 +140,7 @@ def _get_polytrend_timeseries(data: Union[xr.DataArray, xr.Dataset], trend: dict
         polynomial_trend.name = (
             data.name if data.name is not None else "timeseries_polyfit"
         )
-    return polynomial_trend.transpose(*data.dims)
+    return polynomial_trend.transpose(*list(data.dims) + [...])
 
 
 def _subtract_linear_trend(data: Union[xr.DataArray, xr.Dataset], trend: dict):