From d8e2dae6471bd02c8eca62b0bbf9736963146550 Mon Sep 17 00:00:00 2001 From: Scott Staniewicz Date: Tue, 3 Oct 2023 16:09:08 -0700 Subject: [PATCH] `cli_config.py`: add the flag for `--n-parallel-unwrap` --- src/dolphin/workflows/_cli_config.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/dolphin/workflows/_cli_config.py b/src/dolphin/workflows/_cli_config.py index c9570ec6..289f7bef 100644 --- a/src/dolphin/workflows/_cli_config.py +++ b/src/dolphin/workflows/_cli_config.py @@ -29,6 +29,7 @@ def create_config( no_gpu: bool = False, ntiles: tuple[int, int] = (1, 1), downsample_factor: tuple[int, int] = (1, 1), + n_parallel_unwrap: int = 1, use_icu: bool = False, single_update: bool = False, log_file: Optional[Path] = None, @@ -70,6 +71,7 @@ def create_config( unwrap_method=("icu" if use_icu else "snaphu"), ntiles=ntiles, downsample_factor=downsample_factor, + n_parallel_jobs=n_parallel_unwrap, ), worker_settings=dict( block_shape=block_shape, @@ -209,6 +211,12 @@ def get_parser(subparser=None, subcommand_name="run"): " version." ), ) + unwrap_group.add_argument( + "--n-parallel-unwrap", + type=int, + default=1, + help="Number of interferograms to unwrap in parallel.", + ) # Get Outputs from the command line out_group = parser.add_argument_group("Output options")