improve type hint and docstring

viscy/cli.py

@@ -22,7 +22,7 @@ def subcommands() -> dict[str, set[str]]:
         subcommands["export"] = {"model", "dataloaders", "datamodule"}
         return subcommands
 
-    def add_arguments_to_parser(self, parser):
+    def add_arguments_to_parser(self, parser) -> None:
         parser.set_defaults(
             {
                 "trainer.logger": lazy_instance(
@@ -40,16 +40,17 @@ def run_cli(
     model_class: type[LightningModule],
     datamodule_class: type[LightningDataModule],
     trainer_class: type[VisCyTrainer],
-):
+) -> None:
     """
     Main Lightning CLI entry point.
+    Parse log level and set TF32 precision.
 
     Parameters
     ----------
     cli_class : type[LightningCLI]
         Lightning CLI class
     model_class : type[LightningModule]
-        Lightning module class
+        Lightning module class. Not used in preprocess mode.
     datamodule_class : type[LightningDataModule]
         Lightning datamodule class
     trainer_class : type[VisCyTrainer]

viscy/translation/__main__.py

@@ -9,7 +9,7 @@
 class TranslationCLI(VisCyCLI):
     """Extending lightning CLI arguments and defualts."""
 
-    def add_arguments_to_parser(self, parser):
+    def add_arguments_to_parser(self, parser) -> None:
         super().add_arguments_to_parser(parser)
         if "preprocess" not in sys.argv:
             parser.link_arguments("data.yx_patch_size", "model.example_input_yx_shape")