From 1ae5bfc1b8f4bc8971ad8aba04dfce4d24bde37c Mon Sep 17 00:00:00 2001 From: Ellie O'Neil <110510035+eboneil@users.noreply.github.com> Date: Thu, 27 Jun 2024 14:00:22 -0400 Subject: [PATCH] refactor(tags): Use TagUrn class when generating urn (#10786) Co-authored-by: Harshal Sheth --- metadata-ingestion/src/datahub/emitter/mce_builder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/metadata-ingestion/src/datahub/emitter/mce_builder.py b/metadata-ingestion/src/datahub/emitter/mce_builder.py index ccc1267cd9..545cf755a6 100644 --- a/metadata-ingestion/src/datahub/emitter/mce_builder.py +++ b/metadata-ingestion/src/datahub/emitter/mce_builder.py @@ -52,6 +52,7 @@ from datahub.metadata.schema_classes import ( from datahub.utilities.urn_encoder import UrnEncoder from datahub.utilities.urns.data_flow_urn import DataFlowUrn from datahub.utilities.urns.dataset_urn import DatasetUrn +from datahub.utilities.urns.tag_urn import TagUrn logger = logging.getLogger(__name__) Aspect = TypeVar("Aspect", bound=AspectAbstract) @@ -238,8 +239,7 @@ def make_tag_urn(tag: str) -> str: """ if tag and tag.startswith("urn:li:tag:"): return tag - else: - return f"urn:li:tag:{tag}" + return str(TagUrn(tag)) def make_owner_urn(owner: str, owner_type: OwnerType) -> str: