From 8e87d0e3039e5b3c44a9723a273fc13f27f0d2a5 Mon Sep 17 00:00:00 2001 From: Andrej730 Date: Fri, 6 Sep 2024 16:19:02 +0500 Subject: [PATCH] typing --- src/ifcopenshell-python/ifcopenshell/util/attribute.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/ifcopenshell-python/ifcopenshell/util/attribute.py b/src/ifcopenshell-python/ifcopenshell/util/attribute.py index 2fa47589c1..b7a241c5f6 100644 --- a/src/ifcopenshell-python/ifcopenshell/util/attribute.py +++ b/src/ifcopenshell-python/ifcopenshell/util/attribute.py @@ -55,9 +55,9 @@ def get_primitive_type( return "binary" -def get_enum_items(attribute): +def get_enum_items(attribute: ifcopenshell_wrapper.attribute) -> tuple[str, ...]: return attribute.type_of_attribute().declared_type().enumeration_items() -def get_select_items(attribute): +def get_select_items(attribute: ifcopenshell_wrapper.attribute) -> tuple[ifcopenshell_wrapper.entity, ...]: return attribute.type_of_attribute().declared_type().select_list()