release 0.15.4

great_expectations/expectations/registry.py

@@ -9,6 +9,7 @@
 logger = logging.getLogger(__name__)
 
 _registered_expectations = {}
+_registered_data_assistants = {}
 _registered_metrics = {}
 _registered_renderers = {}
 
@@ -94,6 +95,48 @@ def register_expectation(expectation: Type["Expectation"]) -> None:  # noqa: F82
     _registered_expectations[expectation_type] = expectation
 
 
+def register_data_assistant(
+    data_assistant: Type["DataAssistant"],  # noqa: F821
+) -> None:
+    """
+    This method executes "run()" of effective "RuleBasedProfiler" and fills "DataAssistantResult" object with outputs.
+
+    Args:
+        data_assistant: "DataAssistant" class to be registered
+    """
+    data_assistant_type = data_assistant.data_assistant_type
+    if data_assistant_type in _registered_data_assistants:
+        if _registered_data_assistants[data_assistant_type] == data_assistant:
+            logger.info(
+                f'Multiple declarations of DataAssistant "{data_assistant_type}" found.'
+            )
+            return
+        else:
+            logger.warning(
+                f'Overwriting the declaration of DataAssistant "{data_assistant_type}" took place.'
+            )
+
+    logger.debug(
+        f'Registering the declaration of DataAssistant "{data_assistant_type}" took place.'
+    )
+    _registered_data_assistants[data_assistant_type] = data_assistant
+
+
+def get_data_assistant_impl(
+    data_assistant_type: str,
+) -> Optional[Type["DataAssistant"]]:  # noqa: F821
+    """
+    This method obtains (previously registered) "DataAssistant" class from DataAssistant Registry
+
+    Args:
+        data_assistant_type: String representing "snake case" version of "DataAssistant" class type
+
+    Returns:
+        Class inheriting "DataAssistant" if found; otherwise, None
+    """
+    return _registered_data_assistants.get(data_assistant_type)
+
+
 def _add_response_key(res, key, value):
     if key in res:
         res[key].append(value)

great_expectations/rule_based_profiler/data_assistant/data_assistant.py

@@ -5,6 +5,7 @@
 from great_expectations.core import ExpectationSuite
 from great_expectations.core.batch import Batch, BatchRequestBase
 from great_expectations.execution_engine.execution_engine import MetricDomainTypes
+from great_expectations.expectations.registry import register_data_assistant
 from great_expectations.rule_based_profiler.domain_builder import DomainBuilder
 from great_expectations.rule_based_profiler.expectation_configuration_builder import (
     ExpectationConfigurationBuilder,

great_expectations/rule_based_profiler/rule_based_profiler.py

@@ -281,6 +281,7 @@ def run(
 
         rule_state: RuleState
         rule: Rule
+        domains_count: int = 0
         for rule in effective_rules:
             rule_state = rule.run(
                 variables=effective_variables,
@@ -295,6 +296,12 @@ def run(
 
         progress_bar.close()
 
+            domains_count += len(rule_state.domains)
+            progress_bar.update(domains_count)
+            progress_bar.refresh()
+
+        progress_bar.close()
+
     def get_expectation_suite(
         self,
         expectation_suite: Optional[ExpectationSuite] = None,

great_expectations/rule_based_profiler/types/data_assistant_result/data_assistant_result.py

@@ -323,7 +323,11 @@ def get_expect_values_to_be_between_chart(
         )
         anomaly_coded_line = alt.layer(line.layer[0], anomaly_coded_points)
 
-        return band + lower_limit + upper_limit + anomaly_coded_line
+        anomaly_coded_points = line.layer[1].encode(
+            color=point_color_condition, tooltip=tooltip
+        )
+        anomaly_coded_line = alt.layer(line.layer[0], anomaly_coded_points)
+        return anomaly_coded_line
 
     @staticmethod
     def get_interactive_detail_multi_line_chart(

tests/integration/profiling/rule_based_profilers/test_profiler_user_workflows.py

@@ -1752,7 +1752,7 @@ def test_quentin_profiler_user_workflow_multi_batch_quantiles_value_ranges_rule(
                     }
                 ],
                 "rule_count": 1,
-                "variable_count": 6,
+                "variable_count": 5,
             },
             "event": "profiler.run",
             "success": True,

tests/rule_based_profiler/data_assistant/test_volume_data_assistant.py

@@ -23,6 +23,7 @@
     Domain,
     SemanticDomainTypes,
 )
+from great_expectations.rule_based_profiler.types import Domain
 from great_expectations.rule_based_profiler.types.data_assistant_result import (
     DataAssistantResult,
 )