55from aact .nodes import NodeFactory
66
77
8- def test_node_with_non_datamodel_types ():
8+ def test_node_with_non_datamodel_types () -> None :
99 """Test that creating a node with non-DataModel types raises a TypeError."""
1010
1111 class AWrong :
@@ -24,9 +24,9 @@ def __init__(self) -> None:
2424 output_channel_types = [("output" , BWrong )],
2525 )
2626
27- async def event_handler (
27+ async def event_handler ( # type: ignore[override]
2828 self , input_channel : str , input_message : AWrong
29- ) -> None : # type: ignore[override]
29+ ) -> None :
3030 # Handle the event
3131 pass
3232
@@ -40,7 +40,7 @@ async def event_handler(
4040 )
4141
4242
43- def test_node_with_unregistered_datamodel_types ():
43+ def test_node_with_unregistered_datamodel_types () -> None :
4444 """Test that creating a node with unregistered DataModel types raises a TypeError."""
4545
4646 class ACorrect (DataModel ):
@@ -59,9 +59,9 @@ def __init__(self) -> None:
5959 output_channel_types = [("output" , BCorrect )],
6060 )
6161
62- async def event_handler (
62+ async def event_handler ( # type: ignore[override]
6363 self , input_channel : str , input_message : ACorrect
64- ) -> None : # type: ignore[override]
64+ ) -> None :
6565 # Handle the event
6666 pass
6767
@@ -75,7 +75,7 @@ async def event_handler(
7575 )
7676
7777
78- def test_node_with_registered_datamodel_types ():
78+ def test_node_with_registered_datamodel_types () -> None :
7979 """Test that creating a node with properly registered DataModel types succeeds."""
8080
8181 @DataModelFactory .register ("ACorrectNew" )
@@ -96,9 +96,9 @@ def __init__(self) -> None:
9696 output_channel_types = [("output" , BCorrectNew )],
9797 )
9898
99- async def event_handler (
99+ async def event_handler ( # type: ignore[override]
100100 self , input_channel : str , input_message : ACorrectNew
101- ) -> None : # type: ignore[override]
101+ ) -> None :
102102 # Handle the event
103103 pass
104104
0 commit comments