(cls, path: str, arr_name: str)
| 497 | |
| 498 | @classmethod |
| 499 | def load(cls, path: str, arr_name: str): |
| 500 | with open(path, "rb") as f: |
| 501 | arr = np.load(f)[arr_name] |
| 502 | return cls(arr) |
| 503 | |
| 504 | def read_batch(self, batch_size: int) -> Optional[np.ndarray]: |
| 505 | if self.idx >= self.arr.shape[0]: |
no outgoing calls
no test coverage detected