diff --git a/python/pyspark/errors/utils.py b/python/pyspark/errors/utils.py index 16fba7e272bc6..08744dc4c2646 100644 --- a/python/pyspark/errors/utils.py +++ b/python/pyspark/errors/utils.py @@ -174,7 +174,7 @@ def wrapper(*args: Any, **kwargs: Any) -> Any: from pyspark.sql import SparkSession spark = SparkSession.getActiveSession() - if spark is not None: + if spark is not None and hasattr(func, "__name__"): assert spark._jvm is not None pyspark_origin = spark._jvm.org.apache.spark.sql.catalyst.trees.PySparkCurrentOrigin