diff --git a/edvart/report_sections/multivariate_analysis.py b/edvart/report_sections/multivariate_analysis.py index 3ba2459..dcea7e7 100644 --- 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 index 40e5d7a..af44d37 100644 --- 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 += ")"