Learning rate range test.

climax/main/lr_range_test.py

@@ -27,7 +27,7 @@ from climax.main.config import (ERA5_PLEVELS, ERA5_PREDICTORS, PREDICTAND,
                                 CALIB_PERIOD, DOY, SHUFFLE, BATCH_SIZE, OPTIM,
                                 NORM, NET, LOSS, FILTERS, DEM, DEM_FEATURES,
                                 STRATIFY, WET_DAY_THRESHOLD, VALID_SIZE,
-                                ANOMALIES)
+                                ANOMALIES, CHUNKS)
 from climax.main.io import ERA5_PATH, OBS_PATH, DEM_PATH, MODEL_PATH
 
 # module level logger