diff --git a/ingestion/src/metadata/data_quality/builders/i_validator_builder.py b/ingestion/src/metadata/data_quality/builders/i_validator_builder.py index d24df9034178..66cececad6fa 100644 --- a/ingestion/src/metadata/data_quality/builders/i_validator_builder.py +++ b/ingestion/src/metadata/data_quality/builders/i_validator_builder.py @@ -25,7 +25,7 @@ from metadata.generated.schema.tests.testCase import TestCase, TestCaseParameterValue from metadata.generated.schema.type.basic import Timestamp from metadata.profiler.processor.runner import QueryRunner -from metadata.utils.importer import import_test_case_class +from metadata.utils import importer if TYPE_CHECKING: from pandas import DataFrame @@ -59,7 +59,8 @@ def __init__( """ self._test_case = test_case self.runner = runner - self.validator_cls: Type[BaseTestValidator] = import_test_case_class( + # TODO this will be removed on https://github.com/open-metadata/OpenMetadata/pull/18716 + self.validator_cls: Type[BaseTestValidator] = importer.import_test_case_class( entity_type, self._get_source_type(), self.test_case.testDefinition.fullyQualifiedName, # type: ignore