import nit, XValueGen type tweak

single-cell-data · Feb 5, 2025 · 0df1804 · 0df1804
1 parent e0d2bc3
commit 0df1804
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/src/tiledbsoma_ml/query_ids.py b/src/tiledbsoma_ml/query_ids.py
@@ -17,9 +17,7 @@
 
 import numpy as np
 from attrs import define, evolve
-from tiledbsoma import (
-    ExperimentAxisQuery,
-)
+from tiledbsoma import ExperimentAxisQuery
 
 from tiledbsoma_ml._utils import batched, splits
 from tiledbsoma_ml.common import NDArrayJoinId

diff --git a/tests/_utils.py b/tests/_utils.py
@@ -21,7 +21,7 @@
 
 from tiledbsoma_ml.common import MiniBatch, NDArrayJoinId
 
-XValueGen = Callable[[range, range], spmatrix]
+XValueGen = Callable[[range, range], coo_matrix]
 ExpectedBatch = Tuple[List[List[int]], pd.DataFrame]