diff --git a/python/pyspark/testing/connectutils.py b/python/pyspark/testing/connectutils.py index f53ac77b24b6..ee86f5c03974 100644 --- a/python/pyspark/testing/connectutils.py +++ b/python/pyspark/testing/connectutils.py @@ -309,7 +309,7 @@ def _both_conf(): def skip_if_server_version_is( cond: Callable[[LooseVersion], bool], reason: Optional[str] = None -) -> Callable[[...], ...]: +) -> Callable: def decorator(f: Callable) -> Callable: @functools.wraps(f) def wrapper(self, *args, **kwargs): @@ -328,5 +328,5 @@ def wrapper(self, *args, **kwargs): def skip_if_server_version_is_greater_than_or_equal_to( version: str, reason: Optional[str] = None -) -> Callable[[...], ...]: +) -> Callable: return skip_if_server_version_is(lambda v: v >= LooseVersion(version), reason)