Skip to content

Commit a922678

Browse files
committed
Refactor logging initialization and update layout configurations in dependency graph plotting; adjust node size and font size for better visualization
1 parent 8e8f07a commit a922678

File tree

3 files changed

+7
-7
lines changed

3 files changed

+7
-7
lines changed

src/agentlab/experiments/exp_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
import sys
1010
from time import time, sleep
1111

12-
logger = logging.getLogger("agentlab." + __name__) # Get logger based on module name
12+
logger = logging.getLogger(__name__) # Get logger based on module name
1313

1414

1515
# TODO move this to a more appropriate place

src/agentlab/experiments/study.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
from agentlab.experiments.exp_utils import RESULTS_DIR, add_dependencies
1717
from agentlab.experiments.launch_exp import find_incomplete, run_experiments, non_dummy_count
1818

19-
logger = logging.getLogger("agentlab_" + __name__)
19+
logger = logging.getLogger(__name__)
2020

2121

2222
@dataclass

src/agentlab/experiments/view_dep_graph.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -47,9 +47,9 @@ def plot_graph(G, ax, title=None, node_color="lightblue", node_size=40, font_siz
4747
# Use a simple layout for better performance
4848
# pos = nx.spring_layout(G, k=0.1, iterations=100)
4949

50-
# pos = nx.kamada_kawai_layout(G)
50+
pos = nx.kamada_kawai_layout(G)
5151

52-
pos = nx.spectral_layout(G)
52+
# pos = nx.spectral_layout(G)
5353

5454
def name_to_size(name):
5555
if "-" in name:
@@ -83,7 +83,7 @@ def name_to_size(name):
8383

8484

8585
def plot_components_grid(
86-
components, max_cols=4, node_color="lightblue", node_size=1000, font_size=8
86+
components, max_cols=4, node_color="lightblue", node_size=2000, font_size=10
8787
):
8888
"""
8989
Plot components in a grid layout.
@@ -305,8 +305,8 @@ def compress_chains(G):
305305
return G_compressed
306306

307307

308-
benchmark = bgym.DEFAULT_BENCHMARKS["webarena"]()
309-
# benchmark = bgym.DEFAULT_BENCHMARKS["visualwebarena"]()
308+
# benchmark = bgym.DEFAULT_BENCHMARKS["webarena"]()
309+
benchmark = bgym.DEFAULT_BENCHMARKS["visualwebarena"]()
310310

311311
dep_graph = benchmark.dependency_graph_over_tasks()
312312
dep_graph = clean_dict(dep_graph)

0 commit comments

Comments
 (0)