Use __name__ instead of string representation.

ueshin · Mar 5, 2024 · 9fe7f0a · 9fe7f0a
1 parent c417c75
commit 9fe7f0a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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": "<class 'set'>"},
+                message_parameters={"arg_name": "args", "arg_type": "set"},
             )
 
     def test_head(self):