(self, path, max_length=None, surf_path=None, transform=None)
| 198 | class CATH(data.ProteinDataset): |
| 199 | |
| 200 | def __init__(self, path, max_length=None, surf_path=None, transform=None): |
| 201 | path = os.path.expanduser(path) |
| 202 | self.path = path |
| 203 | self.max_length = max_length |
| 204 | |
| 205 | self.pkl_files = sorted([os.path.join(path, f) for f in os.listdir(path) if f.endswith(".pkl")]) |
| 206 | self.transform = transform |
| 207 | if surf_path: |
| 208 | surf_path = os.path.expanduser(surf_path) |
| 209 | self.surf_path = surf_path |
| 210 | |
| 211 | def truncate(self, data_dict, surf_dict=None): |
| 212 | length = data_dict["aatype"].shape[0] |
nothing calls this directly
no outgoing calls
no test coverage detected