diff --git a/climax/core/dataset.py b/climax/core/dataset.py index 83b4dc8ce22505b1279d0a3361b3603ddf72efe0..d385cfb6eb19dc4207b8193ecf7bf5bc4c9f6dba 100644 --- a/climax/core/dataset.py +++ b/climax/core/dataset.py @@ -250,9 +250,7 @@ class EoDataset(torch.utils.data.Dataset): # standardized anomaly = (x(t) - mean(x, t)) / std(x, t) if standard: - anomalies[time] = (anomalies[time] / - ds.isel(time=time_scale).std(dim='time') - ) + anomalies[time] /= ds.isel(time=time_scale).std(dim='time') # concatenate anomalies and sort chronologically anomalies = xr.concat(anomalies.values(), dim='time')