revert datetime aware

open-metadata · Jun 6, 2024 · 455d8e4 · 455d8e4
1 parent f4bf381
commit 455d8e4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/ingestion/tests/unit/topology/pipeline/test_domopipeline.py b/ingestion/tests/unit/topology/pipeline/test_domopipeline.py
@@ -3,7 +3,7 @@
 """
 
 import json
-from datetime import datetime, timezone
+from datetime import datetime
 from pathlib import Path
 from unittest import TestCase
 from unittest.mock import patch
@@ -151,7 +151,7 @@
  sourceUrl=None,
  concurrency=None,
  pipelineLocation=None,
- startDate=datetime(2022, 10, 7, 13, 20, 16, tzinfo=timezone.utc),
+ startDate=datetime(2022, 10, 7, 13, 20, 16),
  tasks=[
  Task(
  name="1",

diff --git a/openmetadata-airflow-apis/tests/unit/ingestion_pipeline/test_workflow_creation.py b/openmetadata-airflow-apis/tests/unit/ingestion_pipeline/test_workflow_creation.py
@@ -79,9 +79,7 @@ def mock_set_ingestion_pipeline_status(self, state):
  return True
 
 
-START_DATE = DateTime(
- datetime.strptime("2022-06-10T15:06:47+00:00", "%Y-%m-%d %H:%M:%S")
-)
+START_DATE = DateTime(datetime.strptime("2022-06-10T15:06:47", "%Y-%m-%d %H:%M:%S"))
 
 
 class OMetaServiceTest(TestCase):