| 55 | |
| 56 | @R.register("datasets.ProteinGym") |
| 57 | class ProteinGym(core.Configurable): |
| 58 | |
| 59 | def __init__(self, path, csv_file): |
| 60 | path = os.path.expanduser(path) |
| 61 | self.path = path |
| 62 | csv_file = os.path.join(path, csv_file) |
| 63 | |
| 64 | with open(csv_file, "r") as f: |
| 65 | reader = csv.DictReader(f) |
| 66 | assay_list = [row for row in reader] |
| 67 | self.ids = [assay["DMS_id"] for assay in assay_list] |
| 68 | self.assay_dict = {assay["DMS_id"]: assay for assay in assay_list} |
| 69 | |
| 70 | |
| 71 | class MutantDataset(torch_data.Dataset): |
nothing calls this directly
no outgoing calls
no test coverage detected