From 70e1ad3fd2174d641e944ca304c34c789f090f46 Mon Sep 17 00:00:00 2001
From: Xinyuan Lin <xinyual3@uci.edu>
Date: Tue, 24 Feb 2026 16:26:16 -0800
Subject: [PATCH] Fix syntax for accessing DataFrame column in
 SklearnTrainingOpDesc

Signed-off-by: Xinyuan Lin <xinyual3@uci.edu>
---
 .../amber/operator/sklearn/training/SklearnTrainingOpDesc.scala | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/sklearn/training/SklearnTrainingOpDesc.scala b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/sklearn/training/SklearnTrainingOpDesc.scala
index c0d1fd3a511..a00cc7ec9c8 100644
--- a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/sklearn/training/SklearnTrainingOpDesc.scala
+++ b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/sklearn/training/SklearnTrainingOpDesc.scala
@@ -98,7 +98,7 @@ class SklearnTrainingOpDesc extends PythonOperatorDescriptor {
        |    def process_table(self, table: Table, port: int) -> Iterator[Optional[TableLike]]:
        |        Y = table[$target]
        |        X = table.drop($target, axis=1)
-       |        X = ${if (countVectorizer) "X[" + text + "]" else "X"}
+       |        X = ${if (countVectorizer) pyb"X[$text]" else "X"}
        |        model = make_pipeline(${if (countVectorizer) "CountVectorizer()," else ""} ${if (
       tfidfTransformer
     ) "TfidfTransformer(),"