Add some TODOs for CLI update

constantinpape · constantinpape · commit 0db92fc0de21 · 2024-11-24T21:15:12.000+01:00
diff --git a/synaptic_reconstruction/tools/cli.py b/synaptic_reconstruction/tools/cli.py
@@ -6,6 +6,7 @@
 
 
 # TODO: handle kwargs
+# TODO: add custom model path
 def segmentation_cli():
     parser = argparse.ArgumentParser(description="Run segmentation.")
     parser.add_argument(
@@ -16,6 +17,7 @@ def segmentation_cli():
         "--output_path", "-o", required=True,
         help="The filepath to directory where the segmentations will be saved."
     )
+    # TODO: list the availabel models here by parsing the keys of the model registry
     parser.add_argument(
         "--model", "-m", required=True, help="The model type."
     )

Original file line number	Diff line number	Diff line change
`@@ -6,6 +6,7 @@`
`6`	`6`
`7`	`7`
`8`	`8`	`# TODO: handle kwargs`
	`9`	`+# TODO: add custom model path`
`9`	`10`	`def segmentation_cli():`
`10`	`11`	`parser = argparse.ArgumentParser(description="Run segmentation.")`
`11`	`12`	`parser.add_argument(`
`@@ -16,6 +17,7 @@ def segmentation_cli():`
`16`	`17`	`"--output_path", "-o", required=True,`
`17`	`18`	`help="The filepath to directory where the segmentations will be saved."`
`18`	`19`	`)`
	`20`	`+ # TODO: list the availabel models here by parsing the keys of the model registry`
`19`	`21`	`parser.add_argument(`
`20`	`22`	`"--model", "-m", required=True, help="The model type."`
`21`	`23`	`)`