Skip to content

Commit

Permalink
Fix CLI for mussels
Browse files Browse the repository at this point in the history
  • Loading branch information
tayden committed Dec 13, 2023
1 parent 8e74a86 commit 3d6ce73
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions kelp_o_matic/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ def find_kelp(
Detect kelp in image at path SOURCE and output the resulting classification raster
to file at path DEST.
"""
find_kelp_(source, dest, species, crop_size, use_nir, tuple(band_order), use_gpu)
find_kelp_(source, dest, species, crop_size, use_nir, band_order, use_gpu)


@cli.command()
Expand All @@ -55,6 +55,12 @@ def find_mussels(
1024,
help="The data window size to run through the segmentation model.",
),
band_order: Optional[list[int]] = typer.Option(
None,
"-b",
help="GDAL-style band re-ordering flag. Defaults to RGB order. "
"To e.g., reorder a BGR image at runtime, pass flags `-b 3 -b 2 -b 1`.",
),
use_gpu: bool = typer.Option(
True, "--gpu/--no-gpu", help="Enable or disable GPU, if available."
),
Expand All @@ -63,7 +69,7 @@ def find_mussels(
Detect mussels in image at path SOURCE and output the resulting classification
raster to file at path DEST.
"""
find_mussels_(source, dest, crop_size, use_gpu)
find_mussels_(source, dest, crop_size, band_order, use_gpu)


def version_callback(value: bool) -> None:
Expand Down

0 comments on commit 3d6ce73

Please sign in to comment.