| 862 | return np.array([sample_info.idx_in_epoch]) |
| 863 | |
| 864 | class SourceClass: |
| 865 | def __init__(self, offset): |
| 866 | self.offset = offset |
| 867 | |
| 868 | @code_smashing_decorator |
| 869 | def __call__(self, sample_info): |
| 870 | return np.array([sample_info.idx_in_epoch + self.offset]) |
| 871 | |
| 872 | class SourceClassWithoutInfo: |
| 873 | def __init__(self, offset): |