typing

src/bonsai/bonsai/bim/module/geometry/operator.py

@@ -211,6 +211,7 @@ class AddRepresentation(bpy.types.Operator, tool.Ifc.Operator):
 
     def _execute(self, context):
         obj = context.active_object
+        assert obj
         props = context.scene.BIMGeometryProperties
         oprops = obj.BIMGeometryProperties
         ifc_context = int(oprops.contexts or "0") or None

src/bonsai/bonsai/bim/module/spatial/data.py

@@ -110,7 +110,7 @@ class SpatialDecompositionData:
                 pass
 
     @classmethod
-    def subelement_class(cls):
+    def subelement_class(cls) -> list[tuple[str, str, str]]:
         results = []
         props = bpy.context.scene.BIMSpatialDecompositionProperties
         if not (container := props.active_container):