From 2e9efe5b828e5d1b03cc96c047fce745ae1d15a6 Mon Sep 17 00:00:00 2001 From: Imri Paran Date: Tue, 26 Nov 2024 16:42:18 +0100 Subject: [PATCH] MINOR: fix: use importer as module (#18793) * fix: use importer as module to allow monkeypatching in collate * Update i_validator_builder.py --- .../metadata/data_quality/builders/i_validator_builder.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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 d24df903417..66cececad6f 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.data_quality.validations.runtime_param_setter.param_setter import 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 @@ class IValidatorBuilder(ABC): """ 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