diff --git a/datahub-actions/src/datahub_actions/plugin/action/metadata_change_sync/metadata_change_sync.py b/datahub-actions/src/datahub_actions/plugin/action/metadata_change_sync/metadata_change_sync.py index 76c5ecdd55..b262851966 100644 --- a/datahub-actions/src/datahub_actions/plugin/action/metadata_change_sync/metadata_change_sync.py +++ b/datahub-actions/src/datahub_actions/plugin/action/metadata_change_sync/metadata_change_sync.py @@ -7,6 +7,7 @@ from pydantic import BaseModel, Field from datahub.emitter.rest_emitter import DatahubRestEmitter from datahub.metadata.schema_classes import ( + ChangeTypeClass, MetadataChangeLogClass, MetadataChangeProposalClass, ) @@ -127,9 +128,12 @@ class MetadataChangeSyncAction(Action): self, orig_event: MetadataChangeLogClass ) -> Union[MetadataChangeProposalClass, None]: try: + changeType = orig_event.get("changeType") + if changeType == ChangeTypeClass.RESTATE or changeType == "RESTATE": + changeType = ChangeTypeClass.UPSERT mcp = MetadataChangeProposalClass( entityType=orig_event.get("entityType"), - changeType=orig_event.get("changeType"), + changeType=changeType, entityUrn=orig_event.get("entityUrn"), entityKeyAspect=orig_event.get("entityKeyAspect"), aspectName=orig_event.get("aspectName"),