Type-annotate cli/run.py

nipy · Mar 29, 2023 · 137e47a · 137e47a
1 parent 0cf5b12
commit 137e47a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/heudiconv/cli/run.py b/heudiconv/cli/run.py
@@ -4,14 +4,15 @@
 import os
 import sys
 from argparse import ArgumentParser
+from typing import Optional
 
 from .. import __version__
 from ..main import workflow
 
 lgr = logging.getLogger(__name__)
 
 
-def main(argv=None):
+def main(argv: Optional[list[str]] = None) -> None:
     parser = get_parser()
     args = parser.parse_args(argv)
     # exit if nothing to be done
@@ -24,7 +25,7 @@ def main(argv=None):
     workflow(**kwargs)
 
 
-def get_parser():
+def get_parser() -> ArgumentParser:
     docstr = ("""Example:
              heudiconv -d 'rawdata/{subject}' -o . -f heuristic.py -s s1 s2 s3""")
     parser = ArgumentParser(description=docstr)