From e46f18f686baa901ecbb654e54684cb10b860f4e Mon Sep 17 00:00:00 2001 From: Andrej730 Date: Wed, 28 Feb 2024 16:24:22 +0500 Subject: [PATCH] type.py - some typing --- src/ifcopenshell-python/ifcopenshell/util/type.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/ifcopenshell-python/ifcopenshell/util/type.py b/src/ifcopenshell-python/ifcopenshell/util/type.py index d5927a016f..3cd2750d2b 100644 --- a/src/ifcopenshell-python/ifcopenshell/util/type.py +++ b/src/ifcopenshell-python/ifcopenshell/util/type.py @@ -19,6 +19,7 @@ import os import json import ifcopenshell.util.schema +from typing import List cwd = os.path.dirname(os.path.realpath(__file__)) @@ -51,11 +52,11 @@ for schema in mapped_schemas: type_to_entity_map[schema][element_type] = [e for e in elements if guessed_element in e] -def get_applicable_types(ifc_class, schema="IFC4"): +def get_applicable_types(ifc_class: str, schema="IFC4") -> List[str]: schema = ifcopenshell.util.schema.get_fallback_schema(schema.upper()) return entity_to_type_map[schema].get(ifc_class, []) -def get_applicable_entities(ifc_type_class, schema="IFC4"): +def get_applicable_entities(ifc_type_class: str, schema="IFC4") -> List[str]: schema = ifcopenshell.util.schema.get_fallback_schema(schema.upper()) return type_to_entity_map[schema].get(ifc_type_class, [])