From 4684aa4a55fa7ed9da9353d9dbf98345cc6fe3a6 Mon Sep 17 00:00:00 2001 From: Dion Moult Date: Wed, 16 Oct 2024 12:26:29 +1100 Subject: [PATCH] Fix bug where you couldn't add conditional material psets --- src/bonsai/bonsai/tool/pset.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/bonsai/bonsai/tool/pset.py b/src/bonsai/bonsai/tool/pset.py index c547330f3c..3faa6e9243 100644 --- a/src/bonsai/bonsai/tool/pset.py +++ b/src/bonsai/bonsai/tool/pset.py @@ -70,10 +70,14 @@ class Pset(bonsai.core.tool.Pset): @classmethod def is_pset_applicable(cls, element: ifcopenshell.entity_instance, pset_name: str) -> bool: + if element.is_a("IfcMaterialDefinition"): + predefined_type = getattr(element, "Category", None) + else: + predefined_type = ifcopenshell.util.element.get_predefined_type(element) return bool( pset_name in bonsai.bim.schema.ifc.psetqto.get_applicable_names( - element.is_a(), ifcopenshell.util.element.get_predefined_type(element), pset_only=True + element.is_a(), predefined_type, pset_only=True ) )