initial cut at a torch.utils.data.IterDataset

single-cell-data · Sep 23, 2024 · 1dde269 · 1dde269
1 parent 566a063
commit 1dde269
Show file tree

Hide file tree

Showing 3 changed files with 1,433 additions and 0 deletions.
diff --git a/src/tiledbsoma_ml/__init__.py b/src/tiledbsoma_ml/__init__.py
@@ -5,4 +5,14 @@
 
 """An API to support machine learning applications built on SOMA."""
 
+from .pytorch import (
+    ExperimentAxisQueryIterableDataset,
+    ExperimentAxisQueryIterDataPipe,
+)
+
 __version__ = "0.1.0-dev"
+
+__all__ = [
+    "ExperimentAxisQueryIterDataPipe",
+    "ExperimentAxisQueryIterableDataset",
+]