Skip to content

Commit 22b9b8f

Browse files
committed
Fix some type annotations in P2P
1 parent 039119e commit 22b9b8f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

sumpy/p2p.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -344,7 +344,7 @@ def __call__(self,
344344
targets: ObjectArray1D[Array] | Array,
345345
sources: ObjectArray1D[Array] | Array,
346346
**kwargs: Any,
347-
) -> Sequence[Array]:
347+
) -> ObjectArray1D[Array]:
348348
knl = self.get_cached_kernel(
349349
targets_is_obj_array=is_obj_array_like(targets),
350350
sources_is_obj_array=is_obj_array_like(sources))
@@ -450,7 +450,7 @@ def __call__(self,
450450
tgtindices: Array,
451451
srcindices: Array,
452452
**kwargs: Any,
453-
) -> tuple[cl.Event, Sequence[Array]]:
453+
) -> ObjectArray1D[Array]:
454454
"""Evaluate a subset of the P2P matrix interactions.
455455
456456
:arg targets: target point coordinates, which can be an object
@@ -810,7 +810,7 @@ def __call__(self,
810810
max_nsources_in_one_box: int,
811811
max_ntargets_in_one_box: int,
812812
**kwargs: Any,
813-
) -> tuple[cl.Event, Sequence[Array]]:
813+
) -> ObjectArray1D[Array]:
814814
from sumpy.array_context import is_cl_cpu
815815

816816
is_gpu = not is_cl_cpu(actx)

0 commit comments

Comments
 (0)