einsumfunc.pyi 737 B

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