| 158 | return noised_x, masked_number |
| 159 | |
| 160 | class BARTDatasetProviderInterface: |
| 161 | def get_shard(self, index, shuffle=True): |
| 162 | raise NotImplementedError |
| 163 | |
| 164 | def release_shard(self, index): |
| 165 | raise NotImplementedError |
| 166 | |
| 167 | def prefetch_shard(self, index): |
| 168 | raise NotImplementedError |
| 169 | |
| 170 | def get_batch(self, batch_iter): |
| 171 | raise NotImplementedError |
| 172 | |
| 173 | def prefetch_batch(self): |
| 174 | raise NotImplementedError |
| 175 | |
| 176 | class BioBARTDatasetProvider(BARTDatasetProviderInterface): |
| 177 | def __init__(self, args): |
nothing calls this directly
no outgoing calls
no test coverage detected