From 889fd53453ebcca4078c34c6fe573a099f721dff Mon Sep 17 00:00:00 2001 From: Ruifeng Zheng Date: Thu, 23 Jan 2025 20:56:48 +0800 Subject: [PATCH] init --- python/pyspark/ml/connect/serialize.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/python/pyspark/ml/connect/serialize.py b/python/pyspark/ml/connect/serialize.py index 8635f22841a75..e93f917e27c96 100644 --- a/python/pyspark/ml/connect/serialize.py +++ b/python/pyspark/ml/connect/serialize.py @@ -18,8 +18,6 @@ import pyspark.sql.connect.proto as pb2 from pyspark.ml.linalg import ( - VectorUDT, - MatrixUDT, DenseVector, SparseVector, DenseMatrix, @@ -184,7 +182,7 @@ def deserialize_param(literal: pb2.Expression.Literal) -> Any: else: raise ValueError(f"Unknown Matrix type {tpe}") else: - raise ValueError(f"Unsupported parameter struct {schema}") + raise ValueError(f"Unknown UDT {jvm_class}") else: return LiteralExpression._to_value(literal)