einsumfunc.pyi 487 B

123456789101112
  1. import numpy as np
  2. import numpy.typing as npt
  3. AR_i: npt.NDArray[np.int64]
  4. AR_f: npt.NDArray[np.float64]
  5. AR_m: npt.NDArray[np.timedelta64]
  6. AR_U: npt.NDArray[np.str_]
  7. np.einsum("i,i->i", AR_i, AR_m) # E: incompatible type
  8. np.einsum("i,i->i", AR_f, AR_f, dtype=np.int32) # E: incompatible type
  9. np.einsum("i,i->i", AR_i, AR_i, out=AR_U) # E: Value of type variable "_ArrayType" of "einsum" cannot be
  10. np.einsum("i,i->i", AR_i, AR_i, out=AR_U, casting="unsafe") # E: No overload variant