You are viewing a plain text version of this content. The canonical link for it is here.
Posted to issues@spark.apache.org by "Sandeep Singh (Jira)" <ji...@apache.org> on 2023/01/05 17:48:00 UTC
[jira] [Created] (SPARK-41903) Support data type ndarray
Sandeep Singh created SPARK-41903:
-------------------------------------
Summary: Support data type ndarray
Key: SPARK-41903
URL: https://issues.apache.org/jira/browse/SPARK-41903
Project: Spark
Issue Type: Sub-task
Components: Connect
Affects Versions: 3.4.0
Reporter: Sandeep Singh
{code:java}
import numpy as np
from pyspark.sql.functions import lit
dtype_to_spark_dtypes = [
(np.int8, [("CAST(1 AS TINYINT)", "tinyint")]),
(np.int16, [("CAST(1 AS SMALLINT)", "smallint")]),
(np.int32, [("CAST(1 AS INT)", "int")]),
(np.int64, [("CAST(1 AS BIGINT)", "bigint")]),
(np.float32, [("CAST(1.0 AS FLOAT)", "float")]),
(np.float64, [("CAST(1.0 AS DOUBLE)", "double")]),
(np.bool_, [("true", "boolean")]),
]
for dtype, spark_dtypes in dtype_to_spark_dtypes:
self.assertEqual(self.spark.range(1).select(lit(dtype(1))).dtypes, spark_dtypes){code}
{code:java}
Traceback (most recent call last):
File "/Users/s.singh/personal/spark-oss/python/pyspark/sql/tests/test_functions.py", line 1064, in test_lit_np_scalar
self.assertEqual(self.spark.range(1).select(lit(dtype(1))).dtypes, spark_dtypes)
File "/Users/s.singh/personal/spark-oss/python/pyspark/sql/utils.py", line 332, in wrapped
return getattr(functions, f.__name__)(*args, **kwargs)
File "/Users/s.singh/personal/spark-oss/python/pyspark/sql/connect/functions.py", line 198, in lit
return Column(LiteralExpression._from_value(col))
File "/Users/s.singh/personal/spark-oss/python/pyspark/sql/connect/expressions.py", line 266, in _from_value
return LiteralExpression(value=value, dataType=LiteralExpression._infer_type(value))
File "/Users/s.singh/personal/spark-oss/python/pyspark/sql/connect/expressions.py", line 262, in _infer_type
raise ValueError(f"Unsupported Data Type {type(value).__name__}")
ValueError: Unsupported Data Type int8
{code}
--
This message was sent by Atlassian Jira
(v8.20.10#820010)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org