MINOR: fix: use importer as module (#18793)

* fix: use importer as module

to allow monkeypatching in collate

* Update i_validator_builder.py
This commit is contained in:
Imri Paran 2024-11-26 16:42:18 +01:00 committed by GitHub
parent 3e7808ba78
commit 2e9efe5b82
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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.tests.testCase import TestCase, TestCaseParameterValue
from metadata.generated.schema.type.basic import Timestamp from metadata.generated.schema.type.basic import Timestamp
from metadata.profiler.processor.runner import QueryRunner from metadata.profiler.processor.runner import QueryRunner
from metadata.utils.importer import import_test_case_class from metadata.utils import importer
if TYPE_CHECKING: if TYPE_CHECKING:
from pandas import DataFrame from pandas import DataFrame
@ -59,7 +59,8 @@ class IValidatorBuilder(ABC):
""" """
self._test_case = test_case self._test_case = test_case
self.runner = runner 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, entity_type,
self._get_source_type(), self._get_source_type(),
self.test_case.testDefinition.fullyQualifiedName, # type: ignore self.test_case.testDefinition.fullyQualifiedName, # type: ignore