From 5926dc7f5b86c3f4bb40158917f49db1be854135 Mon Sep 17 00:00:00 2001 From: sagar-salvi-apptware Date: Tue, 17 Sep 2024 18:11:26 +0530 Subject: [PATCH] fix: Package issue --- .../datahub/ingestion/transformer/add_dataset_ownership.py | 2 +- .../src/datahub/ingestion/transformer/dataset_domain.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/metadata-ingestion/src/datahub/ingestion/transformer/add_dataset_ownership.py b/metadata-ingestion/src/datahub/ingestion/transformer/add_dataset_ownership.py index 1ab3b66f8c49bf..aa7756cf3043a9 100644 --- a/metadata-ingestion/src/datahub/ingestion/transformer/add_dataset_ownership.py +++ b/metadata-ingestion/src/datahub/ingestion/transformer/add_dataset_ownership.py @@ -16,9 +16,9 @@ from datahub.ingestion.transformer.dataset_transformer import ( DatasetOwnershipTransformer, ) -from datahub.metadata._schema_classes import MetadataChangeProposalClass from datahub.metadata.schema_classes import ( BrowsePathsV2Class, + MetadataChangeProposalClass, OwnerClass, OwnershipClass, OwnershipTypeClass, diff --git a/metadata-ingestion/src/datahub/ingestion/transformer/dataset_domain.py b/metadata-ingestion/src/datahub/ingestion/transformer/dataset_domain.py index 984e9d36df7c78..6a838248152650 100644 --- a/metadata-ingestion/src/datahub/ingestion/transformer/dataset_domain.py +++ b/metadata-ingestion/src/datahub/ingestion/transformer/dataset_domain.py @@ -13,11 +13,11 @@ from datahub.ingestion.api.common import PipelineContext from datahub.ingestion.graph.client import DataHubGraph from datahub.ingestion.transformer.dataset_transformer import DatasetDomainTransformer -from datahub.metadata._schema_classes import ( +from datahub.metadata.schema_classes import ( BrowsePathsV2Class, + DomainsClass, MetadataChangeProposalClass, ) -from datahub.metadata.schema_classes import DomainsClass from datahub.utilities.registries.domain_registry import DomainRegistry logger = logging.getLogger(__name__)