Update ma_utils.py

5 days ago · 3742da086d
parent c90d628712
commit 3742da086d
1 changed files with 2 additions and 4 deletions
--- a/swarms/structs/ma_utils.py
+++ b/swarms/structs/ma_utils.py
@ -131,11 +131,9 @@ def set_random_models_for_agents(
        return random.choice(model_names)
    if isinstance(agents, list):
-        return [
+        for agent in agents:
            setattr(agent, "model_name", random.choice(model_names))
-            or agent
+        return agents
            for agent in agents
        ]
    else:
        setattr(agents, "model_name", random.choice(model_names))
        return agents