diff --git a/src/crewai/flow/flow_visual_utils.py b/src/crewai/flow/flow_visual_utils.py index 6296e2627..79220f797 100644 --- a/src/crewai/flow/flow_visual_utils.py +++ b/src/crewai/flow/flow_visual_utils.py @@ -5,7 +5,7 @@ Flow graphs and calculating layout information. These utilities are separated from general-purpose utilities to maintain a clean dependency structure. """ -from typing import Any, Dict, List, Set, TYPE_CHECKING +from typing import TYPE_CHECKING, Any, Dict, List, Set if TYPE_CHECKING: from crewai.flow.flow import Flow diff --git a/src/crewai/flow/visualization_utils.py b/src/crewai/flow/visualization_utils.py index bc44853d0..578f30182 100644 --- a/src/crewai/flow/visualization_utils.py +++ b/src/crewai/flow/visualization_utils.py @@ -4,9 +4,10 @@ import os from pathlib import Path from typing import Any, Callable, Dict, List, Optional, Set, Tuple, cast -from crewai.flow.flow import Flow from pyvis.network import Network +from crewai.flow.flow import Flow + from .core_flow_utils import is_ancestor from .flow_visual_utils import ( build_ancestor_dict,