| 15 | |
| 16 | # define a tmp pytorch dataset |
| 17 | class TmpDataset(Dataset): |
| 18 | def __init__(self, wdir): |
| 19 | self.wdir = wdir |
| 20 | self.mmap = ChunkedMemmap(self.wdir, remove_exist=False) |
| 21 | |
| 22 | def __len__(self): |
| 23 | return len(self.mmap) |
| 24 | |
| 25 | def __getitem__(self, idx): |
| 26 | return np.array(self.mmap[idx]) |
| 27 | |
| 28 | |
| 29 | class MyTestCase(unittest.TestCase): |
no outgoing calls