Make mypy happy again

.github/workflows/integration-test.yml

@@ -41,7 +41,7 @@ jobs:
         run: python -m pip install --upgrade pip
 
       - name: Install Library
-        run: pip install ".[tests]"
+        run: pip install ".[postgresql,tests]"
 
       - name: Setup
         run: |

setup.cfg

@@ -21,6 +21,7 @@ lint =
     mypy==1.14.1
     ruff==0.8.3
     pytest==8.3.4
+    pytest-mock==3.14.0
     great-expectations[spark, spark-connect]>=1.3.1
 gcp =
     great-expectations[gcp]>=1.3.1

tests/integration/test_validate_dataframe_operator.py

@@ -1,8 +1,9 @@
-from typing import Callable
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Callable
 
 import pandas as pd
 import pytest
-from typing import TYPE_CHECKING
 from great_expectations import ExpectationSuite
 from great_expectations.expectations import ExpectColumnValuesToBeInSet
 
@@ -13,7 +14,6 @@
 
 if TYPE_CHECKING:
     from pyspark.sql import SparkSession
-    from pyspark.sql.connect.dataframe import DataFrame as SparkConnectDataFrame
     from pyspark.sql.connect.session import SparkSession as SparkConnectSession
 
 
@@ -90,6 +90,8 @@ def configure_dataframe() -> pyspark.DataFrame:
 
     @pytest.mark.spark_connect_integration
     def test_spark_connect(self, spark_connect_session: SparkConnectSession) -> None:
+        from pyspark.sql.connect.dataframe import DataFrame as SparkConnectDataFrame
+
         column_name = "col_A"
         task_id = f"test_spark_{rand_name()}"
 

tests/unit/test_validate_batch_operator.py

@@ -13,7 +13,6 @@
 from great_expectations.expectations import (
     ExpectColumnValuesToBeInSet,
 )
-from pytest_mock import MockerFixture
 
 from great_expectations_provider.operators.validate_batch import GXValidateBatchOperator