Update test_turn_strategies.py

PrefectHQ · Sep 9, 2024 · 7a9e3cb · 7a9e3cb
1 parent b99677c
commit 7a9e3cb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/orchestration/test_turn_strategies.py b/tests/orchestration/test_turn_strategies.py
@@ -7,7 +7,7 @@
     Popcorn,
     Random,
     RoundRobin,
-    Single,
+    SingleAgent,
 )
 from controlflow.tasks.task import Task
 
@@ -34,7 +34,7 @@ def available_agents(agents: list[Agent], tasks: list[Task]):
 
 
 def test_single_strategy(agents, available_agents):
-    strategy = Single(agent=agents[0])
+    strategy = SingleAgent(agent=agents[0])
     current_agent = agents[0]
 
     tools = strategy.get_tools(current_agent, available_agents)