The lmdb_dataset.py file contains the abstract class for all datasets.
lmdb_dataset.py
The data_interface.py file is used to dynamically load a dataset.
data_interface.py
The saprot folder contains various implementations for loading datasets for SaProt model and ESM models.
saprot