From a79e79ef3dcc577b84fa9c5e17771fb3d4f59c42 Mon Sep 17 00:00:00 2001 From: Pere Miquel Brull Date: Fri, 22 Mar 2024 11:30:02 +0100 Subject: [PATCH] #15662 - List All test cases from a table in DQ (#15665) * #15662 - List All test cases from a table in DQ * #15662 - List All test cases from a table in DQ * #15662 - List All test cases from a table in DQ --- ingestion/src/metadata/data_quality/source/test_suite.py | 4 ++-- .../ingestion/source/database/external_table_lineage_mixin.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/ingestion/src/metadata/data_quality/source/test_suite.py b/ingestion/src/metadata/data_quality/source/test_suite.py index 364eae90585..7ea8056edc0 100644 --- a/ingestion/src/metadata/data_quality/source/test_suite.py +++ b/ingestion/src/metadata/data_quality/source/test_suite.py @@ -83,11 +83,11 @@ class TestSuiteSource(Source): ) -> Optional[List[TestCase]]: """Return test cases if the test suite exists and has them""" if test_suite: - test_cases = self.metadata.list_entities( + test_cases = self.metadata.list_all_entities( entity=TestCase, fields=["testSuite", "entityLink", "testDefinition"], params={"testSuiteId": test_suite.id.__root__}, - ).entities + ) test_cases = cast(List[TestCase], test_cases) # satisfy type checker return test_cases diff --git a/ingestion/src/metadata/ingestion/source/database/external_table_lineage_mixin.py b/ingestion/src/metadata/ingestion/source/database/external_table_lineage_mixin.py index dc07c4357d6..5a1a4e648d4 100644 --- a/ingestion/src/metadata/ingestion/source/database/external_table_lineage_mixin.py +++ b/ingestion/src/metadata/ingestion/source/database/external_table_lineage_mixin.py @@ -13,7 +13,7 @@ External Table Lineage Mixin """ import traceback -from abc import ABC, abstractmethod +from abc import ABC from typing import Iterable from metadata.generated.schema.api.lineage.addLineage import AddLineageRequest