Supporting multiple layers, X, obsm etc. #91
Description
Description of feature
This question came up in #90
Right now, the loader supports one label column and one "data" object. How can we support multiple "data" objects (see #49 for the label column part)?
I can see two routes:
- Inspect the incoming
AnnDataobject to the loader and load whatever is found within that is aligned alongobs(i.e., anyobsm,layers,Xetc.)
ds = ZarrSparseDataset(
batch_size=4096,
chunk_size=32,
preload_nchunks=512,
).add_anndata(my_anndata_with_layers_and_obsm)
next(iter(ds)) # return object contains both `obsm` and `layers`- Expose a RNG argument so two loaders can use RNGs seeded identically. This approach would force the user to create two objects, but would grossly simplify our internal code. I lean towards this because we can always create new APIs that compose two
ZarrSparseDatasetobjects, for example, and handle this internally.