[MAINTENANCE] Update import path for UnexpectedRowsExpectation (#10234)

great-expectations · Aug 20, 2024 · fce8036 · fce8036
1 parent 79694f4
commit fce8036
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/docs/docusaurus/docs/snippets/unexpected_row_expectation.py b/docs/docusaurus/docs/snippets/unexpected_row_expectation.py
@@ -28,7 +28,7 @@
 
 # <snippet name="docs/docusaurus/docs/snippets/unexpected_row_expectation.py define_custom_expectation">
 class UnexpectedTripDistance(UnexpectedRowsExpectation):
-    unexpected_rows_query = """
+    unexpected_rows_query: str = """
         SELECT
             vendor_id, pickup_datetime
         FROM

diff --git a/great_expectations/expectations/__init__.py b/great_expectations/expectations/__init__.py
@@ -1,4 +1,4 @@
-from great_expectations.expectations.expectation import Expectation
+from great_expectations.expectations.expectation import Expectation, UnexpectedRowsExpectation
 
 from .core import (
     ExpectColumnDistinctValuesToBeInSet,

diff --git a/great_expectations/expectations/expectation.py b/great_expectations/expectations/expectation.py
@@ -1613,7 +1613,7 @@ def _validate_metric_value_between(  # noqa: C901, PLR0912
         return {"success": success, "result": {"observed_value": metric_value}}
 
 
-class UnexpectedRowsExpectation(BatchExpectation, ABC):
+class UnexpectedRowsExpectation(BatchExpectation):
     """
     UnexpectedRowsExpectations facilitate the execution of SQL or Spark-SQL queries as the core logic for an Expectation.
 

diff --git a/tests/expectations/test_expectation.py b/tests/expectations/test_expectation.py
@@ -9,11 +9,11 @@
 from great_expectations.compatibility import pydantic
 from great_expectations.data_context.util import file_relative_path
 from great_expectations.exceptions import InvalidExpectationConfigurationError
+from great_expectations.expectations import UnexpectedRowsExpectation
 from great_expectations.expectations.expectation import (
     ColumnMapExpectation,
     ColumnPairMapExpectation,
     MulticolumnMapExpectation,
-    UnexpectedRowsExpectation,
     _validate_dependencies_against_available_metrics,
 )
 from great_expectations.expectations.expectation_configuration import (