From 9fe7f0ab40ffd18a7e429579dfbdd2dd608a99ed Mon Sep 17 00:00:00 2001 From: Takuya UESHIN Date: Tue, 5 Mar 2024 13:36:54 -0800 Subject: [PATCH] Use __name__ instead of string representation. --- python/pyspark/sql/connect/plan.py | 2 +- python/pyspark/sql/session.py | 2 +- python/pyspark/sql/tests/connect/test_connect_basic.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/python/pyspark/sql/connect/plan.py b/python/pyspark/sql/connect/plan.py index ebfe23d16d1b9..7a93580aa1120 100644 --- a/python/pyspark/sql/connect/plan.py +++ b/python/pyspark/sql/connect/plan.py @@ -1182,7 +1182,7 @@ def __init__(self, query: str, args: Optional[Union[Dict[str, Any], List]] = Non elif not isinstance(args, List): raise PySparkTypeError( error_class="INVALID_TYPE", - message_parameters={"arg_name": "args", "arg_type": str(type(args))}, + message_parameters={"arg_name": "args", "arg_type": type(args).__name__}, ) self._query = query diff --git a/python/pyspark/sql/session.py b/python/pyspark/sql/session.py index 53c97b8d43f34..f12a5365d86ca 100644 --- a/python/pyspark/sql/session.py +++ b/python/pyspark/sql/session.py @@ -1687,7 +1687,7 @@ def sql( else: raise PySparkTypeError( error_class="INVALID_TYPE", - message_parameters={"arg_name": "args", "arg_type": str(type(args))}, + message_parameters={"arg_name": "args", "arg_type": type(args).__name__}, ) return DataFrame(self._jsparkSession.sql(sqlQuery, litArgs), self) finally: diff --git a/python/pyspark/sql/tests/connect/test_connect_basic.py b/python/pyspark/sql/tests/connect/test_connect_basic.py index 8aef2d8a85238..cb7e286c5371c 100755 --- a/python/pyspark/sql/tests/connect/test_connect_basic.py +++ b/python/pyspark/sql/tests/connect/test_connect_basic.py @@ -1409,7 +1409,7 @@ def test_sql_with_invalid_args(self): self.check_error( exception=pe.exception, error_class="INVALID_TYPE", - message_parameters={"arg_name": "args", "arg_type": ""}, + message_parameters={"arg_name": "args", "arg_type": "set"}, ) def test_head(self):