From 67a12a04a1502974239dd76134ac03e3c41c8353 Mon Sep 17 00:00:00 2001 From: Michael Davis <36012613+micdavis@users.noreply.github.com> Date: Thu, 26 Jan 2023 10:14:47 -0500 Subject: [PATCH] [BUGFIX] Capital one contrib/micdavis/import hotfix (#6922) Co-authored-by: Anthony Burdi --- .../rule_based_profiler/__init__.py | 0 .../data_profiler_structured_data_assistant.py | 3 ++- .../capitalone_dataprofiler_expectations/tests/conftest.py | 5 +++++ .../test_data_profiler_structured_data_assistant.py | 2 +- 4 files changed, 8 insertions(+), 2 deletions(-) create mode 100644 contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/rule_based_profiler/__init__.py diff --git a/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/rule_based_profiler/__init__.py b/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/rule_based_profiler/__init__.py new file mode 100644 index 000000000000..e69de29bb2d1 diff --git a/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/rule_based_profiler/data_assistant/data_profiler_structured_data_assistant.py b/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/rule_based_profiler/data_assistant/data_profiler_structured_data_assistant.py index 661b289a4e33..740b66ec5086 100644 --- a/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/rule_based_profiler/data_assistant/data_profiler_structured_data_assistant.py +++ b/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/rule_based_profiler/data_assistant/data_profiler_structured_data_assistant.py @@ -1,8 +1,9 @@ from typing import Any, Dict, List, Optional -from contrib.capitalone_dataprofiler_expectations.capitalone_dataprofiler_expectations.rule_based_profiler.data_assistant_result.data_profiler_structured_data_assistant_result import ( +from capitalone_dataprofiler_expectations.rule_based_profiler.data_assistant_result import ( DataProfilerStructuredDataAssistantResult, ) + from great_expectations.rule_based_profiler.config import ParameterBuilderConfig from great_expectations.rule_based_profiler.data_assistant import DataAssistant from great_expectations.rule_based_profiler.data_assistant_result import ( diff --git a/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/tests/conftest.py b/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/tests/conftest.py index ac2d94f1da7e..013cba3e5c19 100644 --- a/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/tests/conftest.py +++ b/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/tests/conftest.py @@ -1,5 +1,10 @@ +import os +import sys + from great_expectations.self_check.util import build_test_backends_list +sys.path.insert(0, os.path.abspath("../..")) + def pytest_addoption(parser): # note: --no-spark will be deprecated in favor of --spark diff --git a/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/tests/rule_based_profiler/data_assistant/test_data_profiler_structured_data_assistant.py b/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/tests/rule_based_profiler/data_assistant/test_data_profiler_structured_data_assistant.py index dff225adf12c..6cc7236f8df1 100644 --- a/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/tests/rule_based_profiler/data_assistant/test_data_profiler_structured_data_assistant.py +++ b/contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/tests/rule_based_profiler/data_assistant/test_data_profiler_structured_data_assistant.py @@ -195,7 +195,7 @@ def test_profile_data_profiler_structured_data_assistant_metrics_count( ): num_metrics += len(parameter_values_for_fully_qualified_parameter_names) - assert num_metrics == 14 + assert num_metrics == 28 @pytest.mark.integration