WIP

python-spec/src/somacore/options.py

@@ -5,7 +5,7 @@
 """
 
 import enum
-from typing import Any, Dict, Mapping, Optional, Sequence, TypeVar, Union
+from typing import Any, Dict, Mapping, Optional, Sequence, TypeVar, Tuple, Union
 
 import attrs
 import numpy as np
@@ -171,6 +171,7 @@ class ResultOrder(enum.Enum):
 
 SparseNDCoord = Union[
     None,
+    Tuple[int,...],
     ValSliceOrSequence[int],
     npt.NDArray[np.integer],
     pa.IntegerArray,

python-spec/src/somacore/types.py

@@ -7,7 +7,8 @@
 
 import sys
 from concurrent import futures
-from typing import TYPE_CHECKING, NoReturn, Optional, Sequence, Type, TypeVar
+from typing import TYPE_CHECKING, Any, NoReturn, Optional, Sequence, Type, TypeVar
+from types import get_original_bases
 
 from typing_extensions import Protocol, TypeGuard
 
@@ -57,6 +58,10 @@ def stop(self) -> Optional[_T_co]: ...
     @property
     def step(self) -> Optional[_T_co]: ...
 
+    @property
+    def __orig_bases__(self) -> Any:
+        return get_original_bases(_T_co)
+
     if sys.version_info < (3, 10) and not TYPE_CHECKING:
         # Python 3.9 and below have a bug where any Protocol with an @property
         # was always regarded as runtime-checkable.