diff --git a/src/ifcopenshell-python/ifcopenshell/api/pset/edit_pset.py b/src/ifcopenshell-python/ifcopenshell/api/pset/edit_pset.py index 2fe9c22c0e..12ee1206cf 100644 --- a/src/ifcopenshell-python/ifcopenshell/api/pset/edit_pset.py +++ b/src/ifcopenshell-python/ifcopenshell/api/pset/edit_pset.py @@ -182,8 +182,7 @@ class Usecase: if self.settings["pset_template"]: self.pset_template = self.settings["pset_template"] else: - # TODO: add IFC2X3 PsetQto template support - self.psetqto = ifcopenshell.util.pset.get_template(self.file.schema) + self.psetqto = ifcopenshell.util.pset.get_template(self.file.schema_identifier) self.pset_template = self.psetqto.get_by_name(self.settings["pset"].Name) def _should_update_prop(self, prop: ifcopenshell.entity_instance) -> bool: diff --git a/src/ifcopenshell-python/ifcopenshell/api/pset/edit_qto.py b/src/ifcopenshell-python/ifcopenshell/api/pset/edit_qto.py index e1bd928101..f23dc593b2 100644 --- a/src/ifcopenshell-python/ifcopenshell/api/pset/edit_qto.py +++ b/src/ifcopenshell-python/ifcopenshell/api/pset/edit_qto.py @@ -140,8 +140,7 @@ class Usecase: if self.settings["pset_template"]: self.pset_template = self.settings["pset_template"] else: - # TODO: add IFC2X3 PsetQto template support - self.psetqto = ifcopenshell.util.pset.get_template("IFC4") + self.psetqto = ifcopenshell.util.pset.get_template(self.file.schema_identifier) self.qto_template = self.psetqto.get_by_name(self.settings["qto"].Name) def update_existing_properties(self):