(self, index)
| 39 | self.sizes = [self.sizes[i] for i in self.avail_idxs] |
| 40 | |
| 41 | def _get_item(self, index): |
| 42 | if hasattr(self, 'avail_idxs') and self.avail_idxs is not None: |
| 43 | index = self.avail_idxs[index] |
| 44 | if self.indexed_ds is None: |
| 45 | self.indexed_ds = IndexedDataset(f'{self.data_dir}/{self.prefix}') |
| 46 | return self.indexed_ds[index] |
| 47 | |
| 48 | def __getitem__(self, index): |
| 49 | hparams = self.hparams |
no test coverage detected