(self, *args, **kwargs)
| 172 | ) |
| 173 | |
| 174 | def generate(self, *args, **kwargs) -> np.ndarray: |
| 175 | return next_and_close(self._generate(*args, streaming=False, **kwargs)) |
| 176 | |
| 177 | def generate_streaming(self, *args, **kwargs) -> Generator[np.ndarray, None, None]: |
| 178 | return self._generate(*args, streaming=True, **kwargs) |