Merge pull request #120 from NCAR/ksha

Bugfix on the production of diagnostic variables in `rollout_to_netcdf.py` + `credit.postblock` update + other minor bugfix
NCAR · Nov 4, 2024 · c6787b6 · c6787b6
2 parents ee70255 + b256574
commit c6787b6
Show file tree

Hide file tree

Showing 22 changed files with 1,010 additions and 244 deletions.
diff --git a/applications/rollout_metrics.py b/applications/rollout_metrics.py
@@ -43,7 +43,7 @@
 from credit.models.checkpoint import load_model_state
 from torch.utils.data import get_worker_info
 from torch.utils.data.distributed import DistributedSampler
-from credit.parser import CREDIT_main_parser, predict_data_check
+from credit.parser import credit_main_parser, predict_data_check
 
 logger = logging.getLogger(__name__)
 warnings.filterwarnings("ignore")
@@ -588,7 +588,7 @@ def predict(rank, world_size, conf, backend, p):
 
     # ======================================================== #
     if conf["data"]["scaler_type"] == "std_new":
-        conf = CREDIT_main_parser(
+        conf = credit_main_parser(
             conf, parse_training=False, parse_predict=True, print_summary=False
         )
         predict_data_check(conf, print_summary=False)