Skip to content

Commit ab53fbd

Browse files
committed
Fix xarray accessor split method repeating same data
1 parent f87feb0 commit ab53fbd

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

dabench/data/_xarray_accessor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@ def split_train_val_test(self,
4848
for sl in split_lengths:
4949
end_i = start_i + sl
5050
out_ds.append(self._obj.isel(time=slice(start_i, end_i)))
51+
start_i = end_i
5152
return tuple(out_ds)
5253

5354

0 commit comments

Comments
 (0)