fix

Signed-off-by: zhilong <[email protected]>

Author: Bye-legumes

python/ray/dag/compiled_dag_node.py

@@ -2650,7 +2650,10 @@ def visualize(
                     "Parameters 'channel_details' are"
                     " not compatible with 'ascii' format."
                 )
-            return self._visualize_ascii()
+            ascii_visualiztion_str = self._visualize_ascii()
+            if view:
+                print(ascii_visualiztion_str)
+            return ascii_visualiztion_str
         try:
             import graphviz
         except ImportError:

python/ray/dag/tests/experimental/test_dag_visualization.py

@@ -7,13 +7,15 @@
 
 import pytest
 
+
 def cleanup_files(filename: str):
     """Helper function to delete files generated during the test."""
-    for ext in ['','.png', '.pdf', '.jpeg', '.dot']:
+    for ext in ["", ".png", ".pdf", ".jpeg", ".dot"]:
         file_path = filename + ext
         if os.path.exists(file_path):
             os.remove(file_path)
 
+
 def test_visualize_basic(ray_start_regular):
     """
     Expect output or dot_source: