use time utility

Author: hsheth2

metadata-ingestion/src/datahub/ingestion/source/dbt.py

@@ -123,6 +123,7 @@
     ViewPropertiesClass,
 )
 from datahub.utilities.mapping import Constants, OperationProcessor
+from datahub.utilities.time import datetime_to_ts_millis
 
 logger = logging.getLogger(__name__)
 DBT_PLATFORM = "dbt"
@@ -1684,7 +1685,7 @@ def get_schema_metadata(
         if node.max_loaded_at is not None:
             actor = mce_builder.make_user_urn("dbt_executor")
             last_modified = AuditStamp(
-                time=int(node.max_loaded_at.timestamp() * 1000),
+                time=datetime_to_ts_millis(node.max_loaded_at),
                 actor=actor,
             )
 

metadata-ingestion/src/datahub/utilities/time.py

@@ -11,6 +11,4 @@ def get_datetime_from_ts_millis_in_utc(ts_millis: int) -> datetime:
 
 
 def datetime_to_ts_millis(dt: datetime) -> int:
-    return int(
-        round(dt.timestamp() * 1000),
-    )
+    return int(round(dt.timestamp() * 1000))