Make sure tasks are always added

graphql-python · Aug 10, 2024 · 76e45ac · 76e45ac
1 parent e1a72ff
commit 76e45ac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/graphql/execution/execute.py b/src/graphql/execution/execute.py
@@ -1504,7 +1504,7 @@ async def await_data() -> None:
                             incremental_data_record, data
                         )
 
-                self.add_task(await_data())  # type: ignore
+                self.add_task(await_data())
 
             else:
                 incremental_publisher.complete_deferred_fragment_record(
@@ -1562,7 +1562,7 @@ async def await_completed_awaitable_item() -> None:
                         incremental_data_record, [value]
                     )
 
-            self.add_task(await_completed_awaitable_item())  # type: ignore
+            self.add_task(await_completed_awaitable_item())
             return incremental_data_record
 
         try:
@@ -1622,6 +1622,7 @@ async def await_completed_item() -> None:
                         incremental_data_record, [value]
                     )
 
+            self.add_task(await_completed_item())
             return incremental_data_record
 
         incremental_publisher.complete_stream_items_record(

diff --git a/src/graphql/execution/incremental_publisher.py b/src/graphql/execution/incremental_publisher.py
@@ -497,7 +497,7 @@ def filter(
                     except AttributeError:
                         pass
                     else:
-                        self._add_task(close_async_iterator)  # type: ignore
+                        self._add_task(close_async_iterator)
 
     def _trigger(self) -> None:
         self._resolve.set()