🎨 optype.io simplifications

Author: jorenham

scipy-stubs/sparse/_matrix_io.pyi

@@ -1,4 +1,4 @@
-from typing import Final, Literal, Protocol, TypeAlias, TypedDict, TypeVar, overload, type_check_only
+from typing import Final, Literal, Protocol, TypeAlias, TypedDict, type_check_only
 
 import optype as op
 from ._bsr import bsr_array, bsr_matrix
@@ -10,12 +10,8 @@ from ._dia import dia_array, dia_matrix
 
 __all__ = ["load_npz", "save_npz"]
 
-_StrOrBytesT = TypeVar("_StrOrBytesT", bound=bytes | str, default=bytes | str)
-_StrOrBytesT_co = TypeVar("_StrOrBytesT_co", bound=bytes | str, default=bytes | str, covariant=True)
-
-@type_check_only
-class _CanWriteBytes(Protocol):
-    def write(self, s: bytes, /) -> object: ...
+_DataArrayOut: TypeAlias = bsr_array | coo_array | csc_array | csr_array | dia_array
+_DataMatrixOut: TypeAlias = bsr_matrix | coo_matrix | csc_matrix | csr_matrix | dia_matrix
 
 @type_check_only
 class _CanReadAndSeekBytes(Protocol):
@@ -26,25 +22,9 @@ class _CanReadAndSeekBytes(Protocol):
 class _PickleKwargs(TypedDict):
     allow_pickle: Literal[False]
 
-# A superior version of `os.PathLike` that's actually valid:
-# Never parametrize generic types with "a TypeVar with constraints"!
-@type_check_only
-class _CanFSPath(Protocol[_StrOrBytesT_co]):
-    @overload
-    def __fspath__(self: _CanFSPath[str], /) -> str: ...
-    @overload
-    def __fspath__(self: _CanFSPath[bytes], /) -> bytes: ...
-    @overload
-    def __fspath__(self: _CanFSPath[bytes | str], /) -> bytes | str: ...
-
-_ToPath: TypeAlias = _StrOrBytesT | _CanFSPath[_StrOrBytesT]
-
-_DataArrayOut: TypeAlias = bsr_array | coo_array | csc_array | csr_array | dia_array
-_DataMatrixOut: TypeAlias = bsr_matrix | coo_matrix | csc_matrix | csr_matrix | dia_matrix
-
 ###
 
 PICKLE_KWARGS: Final[_PickleKwargs] = ...
 
-def load_npz(file: _ToPath | _CanReadAndSeekBytes) -> _DataArrayOut | _DataMatrixOut: ...
-def save_npz(file: _ToPath[str] | _CanWriteBytes, matrix: _data_matrix, compressed: op.CanBool = True) -> None: ...
+def load_npz(file: op.io.ToPath | _CanReadAndSeekBytes) -> _DataArrayOut | _DataMatrixOut: ...
+def save_npz(file: op.io.ToPath[str] | op.io.CanWrite[bytes], matrix: _data_matrix, compressed: op.CanBool = True) -> None: ...