Fix in the default_data_handler.py in multiply_extremes

mlair/data_handler/default_data_handler.py

@@ -273,14 +273,16 @@ class DefaultDataHandler(AbstractDataHandler):
             else:
                 extreme_idx = xr.concat([(Y < -extr_val).any(dim=other_dims[0]),
                                            (Y > extr_val).any(dim=other_dims[0])],
-                                          dim=other_dims[1]).any(dim=other_dims[1])
+                                          dim=other_dims[0]).any(dim=other_dims[0])
 
-            extremes_X = list(map(lambda x: x.sel(**{dim: extreme_idx}), X))
+            sel = extreme_idx[extreme_idx].coords[dim].values
+            extremes_X = list(map(lambda x: x.sel(**{dim: sel}), X))
             self._add_timedelta(extremes_X, dim, timedelta)
             # extremes_X = list(map(lambda x: x.coords[dim].values + np.timedelta64(*timedelta), extremes_X))
 
             extremes_Y = Y.sel(**{dim: extreme_idx})
-            extremes_Y.coords[dim].values += np.timedelta64(*timedelta)
+            #extremes_Y.coords[dim].values += np.timedelta64(*timedelta)
+            self._add_timedelta(extremes_Y, dim, timedelta)
 
             self._Y_extreme = xr.concat([Y, extremes_Y], dim=dim)
             self._X_extreme = list(map(lambda x1, x2: xr.concat([x1, x2], dim=dim), X, extremes_X))