(self, arr_f, shape, dtype)
| 141 | |
| 142 | class StreamingNpzArrayReader(NpzArrayReader): |
| 143 | def __init__(self, arr_f, shape, dtype): |
| 144 | self.arr_f = arr_f |
| 145 | self.shape = shape |
| 146 | self.dtype = dtype |
| 147 | self.idx = 0 |
| 148 | |
| 149 | def read_batch(self, batch_size: int) -> Optional[np.ndarray]: |
| 150 | if self.idx >= self.shape[0]: |
nothing calls this directly
no outgoing calls
no test coverage detected