diff --git a/src/numpy-stubs/__init__.pyi b/src/numpy-stubs/__init__.pyi index 7facd328..8708379f 100644 --- a/src/numpy-stubs/__init__.pyi +++ b/src/numpy-stubs/__init__.pyi @@ -814,7 +814,7 @@ _OrderKACF: TypeAlias = L["K", "A", "C", "F"] | None _FutureScalar: TypeAlias = L["bytes", "str", "object"] _ByteOrderChar: TypeAlias = L["<", ">", "=", "|"] -_CastingKind: TypeAlias = L["no", "equiv", "safe", "same_kind", "unsafe"] +_CastingKind: TypeAlias = L["no", "equiv", "safe", "same_kind", "unsafe", "same_value"] _ModeKind: TypeAlias = L["raise", "wrap", "clip"] _PartitionKind: TypeAlias = L["introselect"] _SortSide: TypeAlias = L["left", "right"] diff --git a/src/numpy-stubs/_core/einsumfunc.pyi b/src/numpy-stubs/_core/einsumfunc.pyi index 158d746f..fd27c45e 100644 --- a/src/numpy-stubs/_core/einsumfunc.pyi +++ b/src/numpy-stubs/_core/einsumfunc.pyi @@ -27,7 +27,7 @@ _ArrayT = TypeVar("_ArrayT", bound=_nt.Array[_nt.co_complex]) # TODO (@jorenham): Annotate the `Sequence` value (numpy/numtype#724) _OptimizeKind: TypeAlias = bool | Literal["greedy", "optimal"] | Sequence[str | tuple[int, ...]] -_CastingSafe: TypeAlias = Literal["no", "equiv", "safe", "same_kind"] +_CastingSafe: TypeAlias = Literal["no", "equiv", "safe", "same_kind", "same_value"] _CastingUnsafe: TypeAlias = Literal["unsafe"] # TODO: Properly handle the `casting`-based combinatorics