From 002c50e29de569d2781fa4bd8dcd6ead104d5f25 Mon Sep 17 00:00:00 2001 From: Andrej730 Date: Tue, 27 Aug 2024 17:31:41 +0500 Subject: [PATCH] black format --- src/bonsai/bonsai/bim/module/spatial/operator.py | 2 +- src/bonsai/bonsai/tool/spatial.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/src/bonsai/bonsai/bim/module/spatial/operator.py b/src/bonsai/bonsai/bim/module/spatial/operator.py index aeaf6033fe..2301c97182 100644 --- a/src/bonsai/bonsai/bim/module/spatial/operator.py +++ b/src/bonsai/bonsai/bim/module/spatial/operator.py @@ -289,7 +289,7 @@ class SelectDecomposedElements(bpy.types.Operator, tool.Ifc.Operator): ifc_class=ifc_class, relating_type=relating_type, is_untyped=is_untyped, - element_filter = element_filter + element_filter=element_filter, ) diff --git a/src/bonsai/bonsai/tool/spatial.py b/src/bonsai/bonsai/tool/spatial.py index d8e6410dfe..df0ae65120 100644 --- a/src/bonsai/bonsai/tool/spatial.py +++ b/src/bonsai/bonsai/tool/spatial.py @@ -257,6 +257,7 @@ class Spatial(bonsai.core.tool.Spatial): keyword: str | None, ) -> filter[ifcopenshell.entity_instance]: keyword = keyword.lower() if keyword else keyword + def filter_element(element): if ifc_class: if not element.is_a(ifc_class): @@ -273,6 +274,7 @@ class Spatial(bonsai.core.tool.Spatial): if keyword not in f"{element.is_a()} {type_name}".lower(): return False return True + return filter(filter_element, elements) @classmethod