From e5e0ccb4f8a12e7a724a60cfef52bf6dec790708 Mon Sep 17 00:00:00 2001
From: Xuan Gu <162244362+xuang7@users.noreply.github.com>
Date: Tue, 24 Feb 2026 19:49:09 -0800
Subject: [PATCH 1/3] update.

---
 .../visualization/bubbleChart/BubbleChartOpDesc.scala      | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/bubbleChart/BubbleChartOpDesc.scala b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/bubbleChart/BubbleChartOpDesc.scala
index 59a8cf5cc89..2dcaa48c09f 100644
--- a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/bubbleChart/BubbleChartOpDesc.scala
+++ b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/bubbleChart/BubbleChartOpDesc.scala
@@ -104,8 +104,13 @@ class BubbleChartOpDesc extends PythonOperatorDescriptor {
 
   def createPlotlyFigure(): PythonTemplateBuilder = {
     assert(xValue.nonEmpty && yValue.nonEmpty && zValue.nonEmpty)
+
+    var isColorEnabled = "False"
+    if (enableColor)
+      isColorEnabled = "True"
+
     pyb"""
-         |        if $enableColor == 'true':
+         |        if $isColorEnabled:
          |            fig = go.Figure(px.scatter(table, x=$xValue, y=$yValue, size=$zValue, size_max=100, color=$colorCategory))
          |        else:
          |            fig = go.Figure(px.scatter(table, x=$xValue, y=$yValue, size=$zValue, size_max=100))

From 9e747c81292a65735abd059130b0d73cc0f1a426 Mon Sep 17 00:00:00 2001
From: Xuan Gu <162244362+xuang7@users.noreply.github.com>
Date: Thu, 26 Feb 2026 14:44:03 -0800
Subject: [PATCH 2/3] update..

---
 .../visualization/bubbleChart/BubbleChartOpDesc.scala      | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/bubbleChart/BubbleChartOpDesc.scala b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/bubbleChart/BubbleChartOpDesc.scala
index 2dcaa48c09f..abf13ae634f 100644
--- a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/bubbleChart/BubbleChartOpDesc.scala
+++ b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/bubbleChart/BubbleChartOpDesc.scala
@@ -104,13 +104,8 @@ class BubbleChartOpDesc extends PythonOperatorDescriptor {
 
   def createPlotlyFigure(): PythonTemplateBuilder = {
     assert(xValue.nonEmpty && yValue.nonEmpty && zValue.nonEmpty)
-
-    var isColorEnabled = "False"
-    if (enableColor)
-      isColorEnabled = "True"
-
     pyb"""
-         |        if $isColorEnabled:
+         |        if '$enableColor' == 'true':
          |            fig = go.Figure(px.scatter(table, x=$xValue, y=$yValue, size=$zValue, size_max=100, color=$colorCategory))
          |        else:
          |            fig = go.Figure(px.scatter(table, x=$xValue, y=$yValue, size=$zValue, size_max=100))

From 89d65b1683c6ce766981c067ff98173393a54949 Mon Sep 17 00:00:00 2001
From: Xuan Gu <162244362+xuang7@users.noreply.github.com>
Date: Mon, 2 Mar 2026 11:22:02 -0800
Subject: [PATCH 3/3] update.

---
 .../operator/visualization/contourPlot/ContourPlotOpDesc.scala  | 2 +-
 .../operator/visualization/ternaryPlot/TernaryPlotOpDesc.scala  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/contourPlot/ContourPlotOpDesc.scala b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/contourPlot/ContourPlotOpDesc.scala
index dd0f41b0faa..939cf8900fe 100644
--- a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/contourPlot/ContourPlotOpDesc.scala
+++ b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/contourPlot/ContourPlotOpDesc.scala
@@ -100,7 +100,7 @@ class ContourPlotOpDesc extends PythonOperatorDescriptor {
        |        y = table[$y].values
        |        z = table[$z].values
        |        grid_size = int($gridSize)
-       |        connGaps = True if $connectGaps == 'true' else False
+       |        connGaps = True if '$connectGaps' == 'true' else False
        |
        |        grid_x, grid_y = np.meshgrid(np.linspace(min(x), max(x), grid_size), np.linspace(min(y), max(y), grid_size))
        |        grid_z = griddata((x, y), z, (grid_x, grid_y), method='cubic')
diff --git a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/ternaryPlot/TernaryPlotOpDesc.scala b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/ternaryPlot/TernaryPlotOpDesc.scala
index 14db98ee20b..b331adb2f26 100644
--- a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/ternaryPlot/TernaryPlotOpDesc.scala
+++ b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/visualization/ternaryPlot/TernaryPlotOpDesc.scala
@@ -101,7 +101,7 @@ class TernaryPlotOpDesc extends PythonOperatorDescriptor {
   /** Returns a Python string that creates the ternary plot figure */
   def createPlotlyFigure(): PythonTemplateBuilder = {
     pyb"""
-       |        if $colorEnabled == 'true' and $colorDataField != "":
+       |        if '$colorEnabled' == 'true' and $colorDataField != "":
        |            fig = px.scatter_ternary(table, a=$firstVariable, b=$secondVariable, c=$thirdVariable, color=$colorDataField)
        |        else:
        |            fig = px.scatter_ternary(table, a=$firstVariable, b=$secondVariable, c=$thirdVariable)