typing

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

@@ -20,7 +20,7 @@ import bpy
 import ifcopenshell.util.element
 import ifcopenshell.util.geolocation
 import ifcopenshell.util.placement
-import ifcopenshell.util.schema
+import ifcopenshell.util.representation
 import ifcopenshell.util.unit
 import bonsai.tool as tool
 from typing import Any, Union

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

@@ -213,9 +213,9 @@ class UnassignGroup(bpy.types.Operator, tool.Ifc.Operator):
 
     def _execute(self, context):
         products = [
-            tool.Ifc.get_entity(o)
+            element
             for o in tool.Blender.get_selected_objects(include_active=False)
-            if tool.Ifc.get_entity(o)
+            if (element := tool.Ifc.get_entity(o))
         ]
         if not products:
             return

src/bonsai/bonsai/bim/module/group/ui.py

@@ -45,6 +45,7 @@ class BIM_PT_groups(Panel):
         if not GroupsData.is_loaded:
             GroupsData.load()
         self.props = tool.Blender.get_group_props()
+        assert self.layout
 
         row = self.layout.row(align=True)
         row.label(text=f"{GroupsData.data['total_groups']} Groups Found", icon="OUTLINER")

src/bonsai/bonsai/bim/module/project/ui.py

@@ -259,6 +259,7 @@ class BIM_PT_project(Panel):
 
     def draw_editable_file_info(self, context):
         pprops = self.props
+        assert self.layout
 
         if ifc_file := tool.Ifc.get():
             row = self.layout.row(align=True)

src/bonsai/bonsai/tool/system.py

@@ -29,7 +29,6 @@ import bonsai.core.tool
 import bonsai.tool as tool
 from bonsai.bim import import_ifc
 import re
-from math import pi, cos, sin
 from mathutils import Matrix, Vector
 from bonsai.bim.module.system.data import ObjectSystemData, SystemDecorationData
 from enum import Enum

src/bonsai/test/tool/test_root.py

@@ -21,6 +21,7 @@ import ifcopenshell
 import ifcopenshell.api
 import ifcopenshell.api.feature
 import ifcopenshell.api.type
+import ifcopenshell.util.element
 import bonsai.core.tool
 import bonsai.tool as tool
 from test.bim.bootstrap import NewFile