ignore function call due to mypy bug described in python/mypy#7203

Wizard-ZP · Jul 28, 2024 · 0296fc7 · 0296fc7
1 parent 0c1cb5f
commit 0296fc7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/paddle/incubate/operators/graph_reindex.py b/python/paddle/incubate/operators/graph_reindex.py
@@ -96,7 +96,7 @@ def graph_reindex(
             >>> neighbors_e1 = paddle.to_tensor(neighbors_e1, dtype="int64")
             >>> count_e1 = paddle.to_tensor(count_e1, dtype="int32")
 
-            >>> reindex_src, reindex_dst, out_nodes = paddle.incubate.graph_reindex(
+            >>> reindex_src, reindex_dst, out_nodes = paddle.incubate.graph_reindex(  # type: ignore[operator]
             ...     x,
             ...     neighbors_e1,
             ...     count_e1,
@@ -118,7 +118,11 @@ def graph_reindex(
 
             >>> neighbors = paddle.concat([neighbors_e1, neighbors_e2])
             >>> count = paddle.concat([count_e1, count_e2])
-            >>> reindex_src, reindex_dst, out_nodes = paddle.incubate.graph_reindex(x, neighbors, count)
+            >>> reindex_src, reindex_dst, out_nodes = paddle.incubate.graph_reindex(  # type: ignore[operator]
+            ...     x,
+            ...     neighbors,
+            ...     count
+            ... )
             >>> print(reindex_src)
             Tensor(shape=[12], dtype=int64, place=Place(cpu), stop_gradient=True,
             [3, 4, 0, 5, 6, 7, 6, 0, 2, 8, 9, 1])