typing

src/ifcpatch/ifcpatch/__init__.py

@@ -25,9 +25,10 @@ import typing
 import inspect
 import collections
 import importlib
+from typing import Union
 
 
-def execute(args):
+def execute(args: dict) -> Union[ifcopenshell.file, str]:
     """Execute a patch recipe
 
     The details of how the patch recipe is executed depends on the definition of
@@ -80,7 +81,7 @@ def execute(args):
     return output
 
 
-def write(output, filepath):
+def write(output: Union[ifcopenshell.file, str], filepath: str) -> None:
     """Write the output of an IFC patch to a file
 
     Typically a patch output would be a patched IFC model file object, or as a