Merge branch 'main' into add-sigma-connector

open-metadata · Sep 25, 2024 · b8f64a7 · b8f64a7
2 parents 2f09364 + d36f01a
commit b8f64a7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ingestion/tests/cli_e2e/common/test_cli_db.py b/ingestion/tests/cli_e2e/common/test_cli_db.py
@@ -18,9 +18,9 @@
 from pathlib import Path
 from typing import Optional
 
-import yaml
 from sqlalchemy.engine import Engine
 
+from metadata.config.common import load_config_file
 from metadata.generated.schema.entity.services.databaseService import DatabaseService
 from metadata.generated.schema.metadataIngestion.workflow import (
     OpenMetadataWorkflowConfig,
@@ -54,7 +54,7 @@ def setUpClass(cls) -> None:
         @classmethod
         def tearDownClass(cls):
             workflow = OpenMetadataWorkflowConfig.model_validate(
-                yaml.safe_load(open(cls.config_file_path))
+                load_config_file(Path(cls.config_file_path))
             )
             db_service: DatabaseService = cls.openmetadata.get_by_name(
                 DatabaseService, workflow.source.serviceName