fix linting

datahub-project · Dec 19, 2024 · d922879 · d922879
1 parent 39ce889
commit d922879
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 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
@@ -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
@@ -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",
                     ),