add output option argument

mllam · Oct 3, 2024 · bb1f8d5 · bb1f8d5
1 parent 0458840
commit bb1f8d5
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 typo in install dependency `distributed` ![\#20](https://github.com/mllam/mllam-data-prep/pull/20)

diff --git a/mllam_data_prep/__main__.py b/mllam_data_prep/__main__.py
@@ -15,6 +15,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"
     )
@@ -61,4 +64,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)