fix: DH-20379: Python scope does not work in format_columns

Author: devonpallison

py/server/deephaven/table.py

@@ -2194,7 +2194,8 @@ def format_columns(self, formulas: Union[str, List[str]]) -> Table:
         """
         try:
             formulas = to_sequence(formulas)
-            return Table(j_table=self.j_table.formatColumns(formulas))
+            with _query_scope_ctx():
+                return Table(j_table=self.j_table.formatColumns(formulas))
         except Exception as e:
             raise DHError(e, "failed to color format columns.") from e
 
@@ -2214,7 +2215,8 @@ def format_column_where(self, col: str, cond: str, formula: str) -> Table:
             DHError
         """
         try:
-            return Table(j_table=self.j_table.formatColumnWhere(col, cond, formula))
+            with _query_scope_ctx():
+                return Table(j_table=self.j_table.formatColumnWhere(col, cond, formula))
         except Exception as e:
             raise DHError(e, "failed to color format column conditionally.") from e
 
@@ -2233,7 +2235,8 @@ def format_row_where(self, cond: str, formula: str) -> Table:
             DHError
         """
         try:
-            return Table(j_table=self.j_table.formatRowWhere(cond, formula))
+            with _query_scope_ctx():
+                return Table(j_table=self.j_table.formatRowWhere(cond, formula))
         except Exception as e:
             raise DHError(e, "failed to color format rows conditionally.") from e