| 870 | return np.array([sample_info.idx_in_epoch + self.offset]) |
| 871 | |
| 872 | class SourceClassWithoutInfo: |
| 873 | def __init__(self, offset): |
| 874 | self.offset = offset |
| 875 | |
| 876 | @code_smashing_decorator |
| 877 | def __call__(self): |
| 878 | return np.array([self.offset]) |
| 879 | |
| 880 | @pipeline_def(batch_size=4, device_id=0, num_threads=4) |
| 881 | def test_pipe(): |