From d92287924ebe2610a86296e5d3744c9e5a32ef93 Mon Sep 17 00:00:00 2001
From: Hyejin Yoon <0327jane@gmail.com>
Date: Thu, 19 Dec 2024 18:43:02 +0900
Subject: [PATCH] fix linting

---
 .../datahub/api/entities/dataprocess/dataprocess_instance.py  | 4 ++--
 .../unit/stateful_ingestion/state/test_stateful_ingestion.py  | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/metadata-ingestion/src/datahub/api/entities/dataprocess/dataprocess_instance.py b/metadata-ingestion/src/datahub/api/entities/dataprocess/dataprocess_instance.py
index d406fa36e00db6..95664463353f89 100644
--- a/metadata-ingestion/src/datahub/api/entities/dataprocess/dataprocess_instance.py
+++ b/metadata-ingestion/src/datahub/api/entities/dataprocess/dataprocess_instance.py
@@ -14,11 +14,11 @@
     RunResultType,
 )
 from datahub.metadata.schema_classes import (
-    AuditStampClass,
     DataProcessInstanceRunEventClass,
     DataProcessInstanceRunResultClass,
     DataProcessRunStatusClass,
     DataProcessTypeClass,
+    TimeStampClass,
 )
 from datahub.utilities.str_enum import StrEnum
 from datahub.utilities.urns.data_flow_urn import DataFlowUrn
@@ -244,7 +244,7 @@ def generate_mcp(
             entityUrn=str(self.urn),
             aspect=DataProcessInstanceProperties(
                 name=self.id,
-                created=AuditStampClass(
+                created=TimeStampClass(
                     time=created_ts_millis or int(time.time() * 1000),
                     actor="urn:li:corpuser:datahub",
                 ),
diff --git a/metadata-ingestion/tests/unit/stateful_ingestion/state/test_stateful_ingestion.py b/metadata-ingestion/tests/unit/stateful_ingestion/state/test_stateful_ingestion.py
index 96ab8f7a01a386..d1b5073bcea2fe 100644
--- a/metadata-ingestion/tests/unit/stateful_ingestion/state/test_stateful_ingestion.py
+++ b/metadata-ingestion/tests/unit/stateful_ingestion/state/test_stateful_ingestion.py
@@ -31,9 +31,9 @@
     DataProcessInstanceProperties,
 )
 from datahub.metadata.schema_classes import (
-    AuditStampClass,
     DataPlatformInstanceClass,
     StatusClass,
+    TimeStampClass,
 )
 from datahub.metadata.urns import DataPlatformUrn, QueryUrn
 from datahub.utilities.urns.dataset_urn import DatasetUrn
@@ -137,7 +137,7 @@ def get_workunits_internal(self) -> Iterable[MetadataWorkUnit]:
                 entityUrn=str(dpi.urn),
                 aspect=DataProcessInstanceProperties(
                     name=dpi.id,
-                    created=AuditStampClass(
+                    created=TimeStampClass(
                         time=int(time.time() * 1000),
                         actor="urn:li:corpuser:datahub",
                     ),