From 3068e42029cc7a45d977742f648a0c0acba4c36c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mart=C3=ADn=20Bravo?=
 <123977407+MartinEBravo@users.noreply.github.com>
Date: Tue, 25 Mar 2025 19:17:35 +0100
Subject: [PATCH] Fix type ignore comment for agent check in get_all_edges
 function

---
 src/agents/extensions/visualization.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/agents/extensions/visualization.py b/src/agents/extensions/visualization.py
index 013a21e..9d0d9f6 100644
--- a/src/agents/extensions/visualization.py
+++ b/src/agents/extensions/visualization.py
@@ -111,7 +111,7 @@ def get_all_edges(agent: Agent, parent: Optional[Agent] = None) -> str:
             "{agent.name}" -> "{handoff.name}";""")
             parts.append(get_all_edges(handoff, agent))
 
-    if not agent.handoffs and not isinstance(agent, Tool):
+    if not agent.handoffs and not isinstance(agent, Tool): # type: ignore
         parts.append(f'"{agent.name}" -> "__end__";')
 
     return "".join(parts)