diff --git a/metadata-ingestion/src/datahub/utilities/source_helpers.py b/metadata-ingestion/src/datahub/utilities/source_helpers.py index b64c053bc93647..2b771aa2a9f756 100644 --- a/metadata-ingestion/src/datahub/utilities/source_helpers.py +++ b/metadata-ingestion/src/datahub/utilities/source_helpers.py @@ -5,7 +5,11 @@ from datahub.ingestion.source.state.stale_entity_removal_handler import ( StaleEntityRemovalHandler, ) -from datahub.metadata.schema_classes import MetadataChangeEventClass, StatusClass +from datahub.metadata.schema_classes import ( + MetadataChangeEventClass, + MetadataChangeProposalClass, + StatusClass, +) from datahub.utilities.urns.urn import guess_entity_type @@ -40,6 +44,9 @@ def auto_status_aspect( elif isinstance(wu.metadata, MetadataChangeProposalWrapper): if isinstance(wu.metadata.aspect, StatusClass): status_urns.add(urn) + elif isinstance(wu.metadata, MetadataChangeProposalClass): + if wu.metadata.aspectName == StatusClass.ASPECT_NAME: + status_urns.add(urn) else: raise ValueError(f"Unexpected type {type(wu.metadata)}")