Fix

open-metadata · Sep 22, 2023 · 69ee476 · 69ee476
1 parent f7fe731
commit 69ee476
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ingestion/tests/unit/topology/database/test_deltalake.py b/ingestion/tests/unit/topology/database/test_deltalake.py
@@ -35,6 +35,7 @@
 )
 from metadata.generated.schema.type.basic import FullyQualifiedEntityName
 from metadata.generated.schema.type.entityReference import EntityReference
+from metadata.ingestion.ometa.ometa_api import OpenMetadata
 from metadata.ingestion.source.database.deltalake.metadata import DeltalakeSource
 
 METASTORE_PATH = "/tmp/spark/unit/metastore"
@@ -108,7 +109,8 @@ class DeltaLakeUnitTest(TestCase):
         MOCK_DELTA_CONFIG
     )
     delta: DeltalakeSource = DeltalakeSource.create(
-        MOCK_DELTA_CONFIG["source"], config.workflowConfig.openMetadataServerConfig
+        MOCK_DELTA_CONFIG["source"],
+        OpenMetadata(config.workflowConfig.openMetadataServerConfig),
     )
     spark = delta.spark