fixed generating pipeline graph when pipeline name not specified

bw2 · Oct 19, 2023 · a7be32a · a7be32a
1 parent 58f699e
commit a7be32a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/step_pipeline/pipeline.py b/step_pipeline/pipeline.py
@@ -219,8 +219,11 @@ def run(self):
 
         args = self.parse_args()
         if args.export_pipeline_graph:
-            output_filename_prefix = re.sub("[:, ]", "_", self.name)
-            output_svg_path = f"{output_filename_prefix}.pipeline_diagram.svg"
+            if self.name is None:
+                output_svg_path = "pipeline_diagram.svg"
+            else:
+                output_filename_prefix = re.sub("[:, ]", "_", self.name)
+                output_svg_path = f"{output_filename_prefix}.pipeline_diagram.svg"
             self.export_pipeline_graph(output_svg_path=output_svg_path)
             print(f"Generated {output_svg_path}. Exiting..")
             sys.exit(0)