fix: add type: ignore for mypy on test fake compiled graph assignments

UIPath-Harshit · claude · UIPath-Harshit · commit a2f166f9f22b · 2026-05-08T01:05:47.000+05:30
Co-Authored-By: Claude Opus 4.6 (1M context) &lt;noreply@anthropic.com&gt;
diff --git a/tests/agent/tools/test_datafabric_tool.py b/tests/agent/tools/test_datafabric_tool.py
@@ -22,7 +22,7 @@ async def test_datafabric_handler_returns_single_terminal_tool_message():
         entity_set=[],
         llm=MagicMock(),
     )
-    handler._compiled = _FakeCompiledGraph(
+    handler._compiled = _FakeCompiledGraph(  # type: ignore[assignment]
         {
             "messages": [
                 ToolMessage(content="{'records': [1], 'total_count': 1}", tool_call_id="1")
@@ -41,7 +41,7 @@ async def test_datafabric_handler_aggregates_multiple_terminal_tool_messages():
         entity_set=[],
         llm=MagicMock(),
     )
-    handler._compiled = _FakeCompiledGraph(
+    handler._compiled = _FakeCompiledGraph(  # type: ignore[assignment]
         {
             "messages": [
                 ToolMessage(
@@ -74,7 +74,7 @@ async def test_datafabric_handler_prefers_terminal_ai_message():
         entity_set=[],
         llm=MagicMock(),
     )
-    handler._compiled = _FakeCompiledGraph(
+    handler._compiled = _FakeCompiledGraph(  # type: ignore[assignment]
         {
             "messages": [
                 ToolMessage(content="{'records': [], 'total_count': 0}", tool_call_id="1"),

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ async def test_datafabric_handler_returns_single_terminal_tool_message():`
`22`	`22`	`entity_set=[],`
`23`	`23`	`llm=MagicMock(),`
`24`	`24`	`)`
`25`		`- handler._compiled = _FakeCompiledGraph(`
	`25`	`+ handler._compiled = _FakeCompiledGraph( # type: ignore[assignment]`
`26`	`26`	`{`
`27`	`27`	`"messages": [`
`28`	`28`	`ToolMessage(content="{'records': [1], 'total_count': 1}", tool_call_id="1")`
`@@ -41,7 +41,7 @@ async def test_datafabric_handler_aggregates_multiple_terminal_tool_messages():`
`41`	`41`	`entity_set=[],`
`42`	`42`	`llm=MagicMock(),`
`43`	`43`	`)`
`44`		`- handler._compiled = _FakeCompiledGraph(`
	`44`	`+ handler._compiled = _FakeCompiledGraph( # type: ignore[assignment]`
`45`	`45`	`{`
`46`	`46`	`"messages": [`
`47`	`47`	`ToolMessage(`
`@@ -74,7 +74,7 @@ async def test_datafabric_handler_prefers_terminal_ai_message():`
`74`	`74`	`entity_set=[],`
`75`	`75`	`llm=MagicMock(),`
`76`	`76`	`)`
`77`		`- handler._compiled = _FakeCompiledGraph(`
	`77`	`+ handler._compiled = _FakeCompiledGraph( # type: ignore[assignment]`
`78`	`78`	`{`
`79`	`79`	`"messages": [`
`80`	`80`	`ToolMessage(content="{'records': [], 'total_count': 0}", tool_call_id="1"),`