This is an automated email from the ASF dual-hosted git repository. linxinyuan pushed a commit to branch aglinxinyuan-patch-2 in repository https://gitbox.apache.org/repos/asf/texera.git
commit 22c04aa88b5508768bdeefad9f109b0b8f5fd408 Author: Xinyuan Lin <[email protected]> AuthorDate: Tue Feb 24 16:15:52 2026 -0800 Fix prediction input shape by squeezing X Signed-off-by: Xinyuan Lin <[email protected]> --- .../texera/amber/operator/sklearn/testing/SklearnTestingOpDesc.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/sklearn/testing/SklearnTestingOpDesc.scala b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/sklearn/testing/SklearnTestingOpDesc.scala index 4c7af2db98..df7d933665 100644 --- a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/sklearn/testing/SklearnTestingOpDesc.scala +++ b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/sklearn/testing/SklearnTestingOpDesc.scala @@ -69,7 +69,7 @@ class SklearnTestingOpDesc extends PythonOperatorDescriptor { | table = Table(self.data) | Y = table[$target] | X = table.drop($target, axis=1) - | predictions = model.predict(X) + | predictions = model.predict(X.squeeze()) | if $isRegressionStr: | tuple_["R2"] = r2_score(Y, predictions) | tuple_["RMSE"] = root_mean_squared_error(Y, predictions)
