Merge pull request #205 from lilab-bcb/typing

inline index types
lilab-bcb · Sep 29, 2023 · 2c00357 · 2c00357
2 parents 6347bc1 + 4a7adf6
commit 2c00357
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/cirrocumulus/sparse_dataset.py b/cirrocumulus/sparse_dataset.py
@@ -13,7 +13,6 @@
 
 import numpy as np
 import scipy.sparse as ss
-from anndata.compat import Index1D
 from scipy.sparse import _sparsetools
 
 
@@ -23,7 +22,11 @@
 except ImportError:
     _cs_matrix = ss.spmatrix
 
-from anndata._core.index import Index, _subset
+from anndata._core.index import _subset
+
+
+Index1D = Union[slice, int, str, np.int64, np.ndarray]
+Index = Union[Index1D, Tuple[Index1D, Index1D], ss.spmatrix]
 
 
 def unpack_index(index: Index) -> tuple[Index1D, Index1D]: