(
self,
data_source,
split,
subsample
)
| 18 | |
| 19 | class SDFSamples(torch.utils.data.Dataset): |
| 20 | def __init__( |
| 21 | self, |
| 22 | data_source, |
| 23 | split, |
| 24 | subsample |
| 25 | ): |
| 26 | self.subsample = subsample |
| 27 | self.data_source = data_source |
| 28 | self.npyfiles = get_instance_filenames(data_source, split) |
| 29 | |
| 30 | |
| 31 | def __len__(self): |
nothing calls this directly
no test coverage detected