add output option argument (#26)

mllam · Oct 30, 2024 · a283eff · a283eff
1 parent 077faa5
commit a283eff
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## vX.X.X
 
+### Added
+
+- add optional output path argument to parser. ![\#26](https://github.com/mllam/mllam-data-prep/pull/26)
+
 ### Changed
 
 - fix bug by making dependency `distributed` optional ![\#27](https://github.com/mllam/mllam-data-prep/pull/27)

diff --git a/mllam_data_prep/__main__.py b/mllam_data_prep/__main__.py
@@ -21,6 +21,9 @@
         formatter_class=argparse.ArgumentDefaultsHelpFormatter
     )
     parser.add_argument("config", help="Path to the config file", type=Path)
+    parser.add_argument(
+        "-o", "--output", help="Path to the output zarr file", type=Path, default=None
+    )
     parser.add_argument(
         "--show-progress", help="Show progress bar", action="store_true"
     )
@@ -74,4 +77,4 @@
         # print the dashboard link
         logger.info(f"Dashboard link: {cluster.dashboard_link}")
 
-    create_dataset_zarr(fp_config=args.config)
+    create_dataset_zarr(fp_config=args.config, fp_zarr=args.output)