Add __getitem__ to base dataset class

BorgwardtLab / proteinshake

Protein structure datasets for machine learning.

https://proteinshake.ai

BSD 3-Clause "New" or "Revised" License

99 stars 8 forks source link

Open timkucera opened 1 year ago

timkucera commented 1 year ago

to allow:

from proteinshake.datasets import AlphaFoldDataset
ds = AlphaFoldDataset(...)
print(ds[0])
# instead of: proteins, size = ds.proteins()

timkucera commented 1 year ago

also to tasks

timkucera commented 1 year ago

and add __len__ to return size