diff --git a/metagpt/ext/sela/scripts/visualize_experiment.py b/metagpt/ext/sela/scripts/visualize_experiment.py
index 4133942fc..a6d980d11 100644
--- a/metagpt/ext/sela/scripts/visualize_experiment.py
+++ b/metagpt/ext/sela/scripts/visualize_experiment.py
@@ -1,6 +1,9 @@
 import networkx as nx
 
-from metagpt.ext.sela.evaluation.visualize_mcts import build_tree_recursive, visualize_tree
+from metagpt.ext.sela.evaluation.visualize_mcts import (
+    build_tree_recursive,
+    visualize_tree,
+)
 from metagpt.ext.sela.MCTS import MCTS, create_initial_state, initialize_di_root_node
 from metagpt.ext.sela.run_experiment import get_args
 from metagpt.ext.sela.utils import DATA_CONFIG