typing

src/ifcopenshell-python/ifcopenshell/api/root/create_entity.py

@@ -18,10 +18,17 @@
 
 import ifcopenshell
 import ifcopenshell.api
+from typing import Optional
 
 
 class Usecase:
-    def __init__(self, file, ifc_class="IfcBuildingElementProxy", predefined_type=None, name=None):
+    def __init__(
+        self,
+        file: ifcopenshell.file,
+        ifc_class: str = "IfcBuildingElementProxy",
+        predefined_type: Optional[str] = None,
+        name: Optional[str] = None,
+    ):
         """Create a new rooted product
 
         This is a critical function used to create almost any rooted product or
@@ -70,7 +77,7 @@ class Usecase:
             "name": name,
         }
 
-    def execute(self):
+    def execute(self) -> ifcopenshell.entity_instance:
         element = self.file.create_entity(
             self.settings["ifc_class"],
             **{

src/ifcopenshell-python/ifcopenshell/api/root/remove_product.py

@@ -21,7 +21,7 @@ import ifcopenshell.util.element
 
 
 class Usecase:
-    def __init__(self, file: ifcopenshell.entity_instance, product: ifcopenshell.entity_instance):
+    def __init__(self, file: ifcopenshell.file, product: ifcopenshell.entity_instance):
         """Removes a product
 
         This is effectively a smart delete function that not only removes a