opt distributed example

modelscope · Feb 1, 2024 · 573dedc · 573dedc
1 parent 68ecafd
commit 573dedc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -88,6 +88,7 @@ repos:
           --disable=W1113,
           --disable=W0221,
           --disable=R0401,
+          --disable=W0632,
         ]
   - repo: https://github.com/regebro/pyroma
     rev: "4.0"

diff --git a/examples/distributed/distributed_debate.py b/examples/distributed/distributed_debate.py
@@ -85,13 +85,10 @@ def setup_server(parsed_args: argparse.Namespace) -> None:
 
 def run_main_process(parsed_args: argparse.Namespace) -> None:
     """Setup the main debate competition process"""
-    agents = agentscope.init(
+    pro_agent, con_agent, judge_agent = agentscope.init(
         model_configs="configs/model_configs.json",
         agent_configs="configs/debate_agent_configs.json",
     )
-    pro_agent = agents[0]
-    con_agent = agents[1]
-    judge_agent = agents[2]
     pro_agent = pro_agent.to_dist(
         host=parsed_args.pro_host,
         port=parsed_args.pro_port,