From 224bdc3a4036e669b2abceec9fee5e4110db9f7f Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Tue, 21 Nov 2023 11:48:52 -0600 Subject: [PATCH] _binary_op: precisely type op --- pytato/array.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytato/array.py b/pytato/array.py index 0f9fc5d..52a7283 100644 --- a/pytato/array.py +++ b/pytato/array.py @@ -526,7 +526,7 @@ class Array(Taggable): __rmatmul__ = partialmethod(__matmul__, reverse=True) def _binary_op(self, - op: Any, + op: Callable[[ScalarExpression, ScalarExpression], ScalarExpression], other: ArrayOrScalar, get_result_type: Callable[[DtypeOrScalar, DtypeOrScalar], np.dtype[Any]] = _np_result_type, # noqa reverse: bool = False) -> Array: -- GitLab