Make arg optional

OCHA-DAP · Oct 23, 2024 · 6cf7ab0 · 6cf7ab0
1 parent de93b78
commit 6cf7ab0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/run_raster_stats.py b/run_raster_stats.py
@@ -116,16 +116,10 @@ def process_chunk(start, end, dataset, mode, df_iso3s, engine_url):
 
 if __name__ == "__main__":
     args = cli_args()
-    dataset = args.dataset
-    logger.info(f"Updating data for {dataset}...")
 
     engine_url = db_engine_url(args.mode)
     engine = create_engine(engine_url)
 
-    create_qa_table(engine)
-    settings = load_pipeline_config(dataset)
-    start, end, is_forecast = parse_pipeline_config(settings, args.test)
-    create_dataset_table(dataset, engine, is_forecast)
     if args.update_metadata:
         logger.info("Updating metadata in Postgres database...")
         create_iso3_df(engine)
@@ -137,6 +131,14 @@ def process_chunk(start, end, dataset, mode, df_iso3s, engine_url):
         )
         sys.exit(0)
 
+    dataset = args.dataset
+    logger.info(f"Updating data for {dataset}...")
+
+    create_qa_table(engine)
+    settings = load_pipeline_config(dataset)
+    start, end, is_forecast = parse_pipeline_config(settings, args.test)
+    create_dataset_table(dataset, engine, is_forecast)
+
     sel_iso3s = settings["test"]["iso3s"] if args.test else None
     df_iso3s = get_iso3_data(sel_iso3s, engine)
     date_ranges = split_date_range(start, end)

diff --git a/src/utils/inputs.py b/src/utils/inputs.py
@@ -11,6 +11,7 @@ def cli_args():
         help="Dataset for which to calculate raster stats",
         choices=["seas5", "era5", "imerg"],
         default=None,
+        nargs="?",
     )
     parser.add_argument(
         "--mode",