From 6cf7ab0d148b6661b465057f27a3bb8836d03cbf Mon Sep 17 00:00:00 2001
From: hannahker <hannah.ker96@gmail.com>
Date: Wed, 23 Oct 2024 12:53:00 -0400
Subject: [PATCH] Make  arg optional

---
 run_raster_stats.py | 14 ++++++++------
 src/utils/inputs.py |  1 +
 2 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/run_raster_stats.py b/run_raster_stats.py
index 1ca2906..41b8da3 100644
--- 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
index 907907a..0677d8a 100644
--- 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",