Fix the bug where utilsd tries to add NoneType arguments to argparse (#10)

Author: victorywys

tests/test_config_cli.py

@@ -1,5 +1,5 @@
 import os
-from typing import Optional
+from typing import Optional, Union
 
 from utilsd.config import PythonConfig, Registry, RegistryConfig, SubclassConfig, configclass
 from unittest.mock import patch
@@ -38,9 +38,10 @@ class TEST(metaclass=Registry, name="test"):
 
 @TEST.register_module()
 class Module1:
-    def __init__(self, a: int = 1, b: Optional[str] = None):
+    def __init__(self, a: int = 1, b: Optional[str] = None, c: Union[str, BaseFoo, None] = None):
         self.a = a
         self.b = b
+        self.c = c
 
 
 @configclass
@@ -100,4 +101,5 @@ def test_registry_config_command_line():
 
 
 if __name__ == '__main__':
-    test_cli_with_bool()
+    # test_cli_with_bool()
+    test_registry_config_command_line()

utilsd/config/cli_parser.py

@@ -60,7 +60,10 @@ def build_parser(self, parser: ArgumentParser,
             if not isinstance(shortcut, list):
                 raise TypeError(f'Shortcut of {name} is not found to be a list: {shortcut}')
 
-            if issubclass(type_, Enum):
+            if type_ is type(None):
+                # ignore None type when building parser
+                pass
+            elif issubclass(type_, Enum):
                 parser.add_argument('--' + name, *shortcut, dest=name, type=str, metavar='STRING',
                                     default=SUPPRESS, choices=[e.value for e in type_])
             elif type_ in (int, str, float, bool, list, dict):