This is an automated email from the ASF dual-hosted git repository. linxinyuan pushed a commit to branch xinyuan-fix-sklearn-testing in repository https://gitbox.apache.org/repos/asf/texera.git
commit 08d35242c654a29be36b4c89108248374d75b601 Author: Xinyuan Lin <[email protected]> AuthorDate: Tue Feb 24 16:07:53 2026 -0800 init --- .../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)
