From 2ad5d6f7e1d85da373f8bf56aa3a6d74e0e65759 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 19 Oct 2021 11:09:47 +0200
Subject: [PATCH] Consistent chunk size.
---
climax/main/downscale_infer.py | 2 +-
climax/main/downscale_train.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/climax/main/downscale_infer.py b/climax/main/downscale_infer.py
index 0a83e0f..2918217 100644
--- a/climax/main/downscale_infer.py
+++ b/climax/main/downscale_infer.py
@@ -75,7 +75,7 @@ if __name__ == '__main__':
dem = dem.drop_vars(['slope', 'aspect'])
# add dem to set of predictor variables
- Era5_ds = xr.merge([Era5_ds, dem])
+ Era5_ds = xr.merge([Era5_ds, dem]).chunk(Era5_ds.chunks)
# load pretrained model
if state_file.exists():
diff --git a/climax/main/downscale_train.py b/climax/main/downscale_train.py
index 7468fa8..1535759 100644
--- a/climax/main/downscale_train.py
+++ b/climax/main/downscale_train.py
@@ -101,7 +101,7 @@ if __name__ == '__main__':
# check whether to use slope and aspect
if not DEM_FEATURES:
- dem = dem.drop_vars(['slope', 'aspect'])
+ dem = dem.drop_vars(['slope', 'aspect']).chunk(Era5_ds.chunks)
# add dem to set of predictor variables
Era5_ds = xr.merge([Era5_ds, dem])
--
GitLab