update to somacore>=1.0.24, tiledbsoma>=1.15.0rc4

single-cell-data · Dec 2, 2024 · 643b328 · 643b328
1 parent 2e9e5b1
commit 643b328
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -7,7 +7,8 @@ name = "tiledbsoma-ml"
 dynamic = ["version"]
 dependencies = [
     "attrs>=22.2",
-    "tiledbsoma>=1.9.0",
+    "somacore>=1.0.24",
+    "tiledbsoma>=1.15.0rc4",
     "torch>=2.0",
     "torchdata<=0.9",
     "numpy",

diff --git a/src/tiledbsoma_ml/pytorch.py b/src/tiledbsoma_ml/pytorch.py
@@ -41,7 +41,7 @@
 import tiledbsoma as soma
 import torch
 import torchdata
-from somacore.query._eager_iter import EagerIterator as _EagerIterator
+from tiledbsoma._eager_iter import EagerIterator as _EagerIterator
 from typing_extensions import Self
 
 logger = logging.getLogger("tiledbsoma_ml.pytorch")

diff --git a/tests/test_pytorch.py b/tests/test_pytorch.py
@@ -100,6 +100,7 @@ def add_dataframe(coll: CollectionBase, key: str, value_range: range) -> None:
             ]
         ),
         index_column_names=["soma_joinid"],
+        domain=((value_range.start, value_range.stop),),
     )
     df.write(
         pa.Table.from_pydict(