Minor typing fixes

childmindresearch · Jul 12, 2024 · e6b0617 · e6b0617
1 parent 28bc3a9
commit e6b0617
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/styx/compiler/compile/descriptor.py b/src/styx/compiler/compile/descriptor.py
@@ -39,7 +39,7 @@ def _generate_run_function(
 
     # Function arguments
     func.args.extend(build_input_arguments(inputs, sub_aliases))
-    func.args.append(PyArg(name="runner", type="Runner", default="None", docstring="Command runner"))
+    func.args.append(PyArg(name="runner", type="Runner | None", default="None", docstring="Command runner"))
 
     # Function body: Runner instantiation
     func.body.extend([

diff --git a/src/styx/compiler/compile/inputs.py b/src/styx/compiler/compile/inputs.py
@@ -34,12 +34,13 @@ def _base() -> str:
             case _:
                 assert False
 
+    type_ = _base()
     if arg.type.primitive != InputTypePrimitive.Flag:
         if arg.type.is_list:
-            return f"list[{_base()}]"
+            type_ = f"list[{type_}]"
         if arg.type.is_optional:
-            return f"{_base()} | None"
-    return _base()
+            type_ = f"{type_} | None"
+    return type_
 
 
 def build_input_arguments(