diff --git a/arraycontext/fake_numpy.py b/arraycontext/fake_numpy.py index 5645902b..f5370256 100644 --- a/arraycontext/fake_numpy.py +++ b/arraycontext/fake_numpy.py @@ -413,6 +413,10 @@ def logaddexp(self, a: ArrayOrContainerOrScalarT, / ) -> ArrayOrContainerOrScalarT: ... def logaddexp2(self, a: ArrayOrContainerOrScalarT, / ) -> ArrayOrContainerOrScalarT: ... + def real(self, a: ArrayOrContainerOrScalarT, /, + ) -> ArrayOrContainerOrScalarT: ... + def imag(self, a: ArrayOrContainerOrScalarT, /, + ) -> ArrayOrContainerOrScalarT: ... # {{{ binary/ternary ufuncs # FIXME: These are more restrictive than necessary, but they'll do the job