Fix DEM processing.

climax/main/downscale_infer.py

@@ -65,6 +65,10 @@ if __name__ == '__main__':
             dem, {'y': Era5_ds.y, 'x': Era5_ds.x},
             add_coord={'time': Era5_ds.time})
 
+        # check whether to use slope and aspect
+        if not DEM_FEATURES:
+            dem = dem.drop_vars(['slope', 'aspect'])
+
         # add dem to set of predictor variables
         Era5_ds = xr.merge([Era5_ds, dem])
 

climax/main/downscale_train.py

@@ -90,7 +90,7 @@ if __name__ == '__main__':
                 dem, {'y': Era5_ds.y, 'x': Era5_ds.x},
                 add_coord={'time': Era5_ds.time})
 
-            # check wether to use slope and aspect
+            # check whether to use slope and aspect
             if not DEM_FEATURES:
                 dem = dem.drop_vars(['slope', 'aspect'])