support somacore>=1.0.24 / tiledbsoma>=1.15.0rc4

.github/workflows/python-tilledbsoma-ml-compat.yml

@@ -31,6 +31,7 @@ jobs:
           - "tiledbsoma~=1.12.0"
           - "tiledbsoma~=1.13.0"
           - "tiledbsoma~=1.14.0"
+          - "tiledbsoma~=1.15.0rc4"
 
     runs-on: ${{ matrix.os }}
 

src/tiledbsoma_ml/pytorch.py

@@ -41,7 +41,13 @@
 import tiledbsoma as soma
 import torch
 import torchdata
-from somacore.query._eager_iter import EagerIterator as _EagerIterator
+
+try:
+    # somacore<1.0.24 / tiledbsoma<1.15
+    from somacore.query._eager_iter import EagerIterator as _EagerIterator
+except ImportError:
+    # somacore>=1.0.24 / tiledbsoma>=1.15
+    from tiledbsoma._eager_iter import EagerIterator as _EagerIterator
 from typing_extensions import Self
 
 logger = logging.getLogger("tiledbsoma_ml.pytorch")

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(