fix: Package issue

sagar-salvi-apptware · Sep 17, 2024 · 5926dc7 · 5926dc7
1 parent 60ebc07
commit 5926dc7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 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
@@ -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
@@ -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__)