add title for pipe plots (#1162)

aimclub · Sep 4, 2023 · b1f8bd6 · b1f8bd6
1 parent 1a96f38
commit b1f8bd6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fedot/core/pipelines/pipeline.py b/fedot/core/pipelines/pipeline.py
@@ -409,11 +409,11 @@ def replace_n_jobs_in_nodes(self, n_jobs: int):
     def show(self, save_path: Optional[Union[PathLike, str]] = None, engine: Optional[str] = None,
              node_color: Optional[NodeColorType] = None, dpi: Optional[int] = None,
              node_size_scale: Optional[float] = None, font_size_scale: Optional[float] = None,
-             edge_curvature_scale: Optional[float] = None,
+             edge_curvature_scale: Optional[float] = None, title: Optional[str] = None,
              nodes_labels: Dict[int, str] = None, edges_labels: Dict[int, str] = None):
         PipelineVisualizer(self).visualise(save_path=save_path, engine=engine, node_color=node_color,
                                            dpi=dpi, node_size_scale=node_size_scale, font_size_scale=font_size_scale,
-                                           nodes_labels=nodes_labels, edges_labels=edges_labels)
+                                           title=title, nodes_labels=nodes_labels, edges_labels=edges_labels)
 
 
 def _graph_nodes_to_pipeline_nodes(operator: LinkedGraph, nodes: Sequence[PipelineNode]):