fix: missing quotes in exported code

datamole-ai · Jul 27, 2023 · 1c16ec6 · 1c16ec6
1 parent 71ac5c1
commit 1c16ec6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/edvart/report_sections/multivariate_analysis.py b/edvart/report_sections/multivariate_analysis.py
@@ -227,7 +227,7 @@ def add_cells(self, cells: List[Dict[str, Any]]) -> None:
             if self.columns is not None:
                 code += f", columns={self.columns}"
             if self.color_col is not None:
-                code += f", color_col={self.color_col}"
+                code += f", color_col='{self.color_col}'"
             code += ")"
             cells.append(nbfv4.new_code_cell(code))
             for sub in self.subsections:

diff --git a/edvart/report_sections/timeseries_analysis/time_analysis_plot.py b/edvart/report_sections/timeseries_analysis/time_analysis_plot.py
@@ -182,7 +182,7 @@ def add_cells(self, cells: List[Dict[str, Any]]) -> None:
         if self.columns is not None:
             default_call += f", columns={self.columns}"
         if self.color_col is not None:
-            default_call += f", color_col={self.color_col}"
+            default_call += f", color_col='{self.color_col}'"
         if self.separate_plots:
             default_call += ", separate_plots=True"
         default_call += ")"