(self, path: str, model: str, batch_size: int)
| 56 | """ |
| 57 | |
| 58 | def __init__(self, path: str, model: str, batch_size: int) -> None: |
| 59 | self._path = path |
| 60 | self._data = BatchManifest( |
| 61 | version=1, |
| 62 | model=model, |
| 63 | batch_size=batch_size, |
| 64 | total_batches=None, |
| 65 | batches=[], |
| 66 | ) |
| 67 | self._lock = threading.Lock() |
| 68 | os.makedirs(os.path.dirname(path), exist_ok=True) |
| 69 | |
| 70 | def set_total_batches(self, n: int) -> None: |
| 71 | """Set the total number of batches (called once after grouping).""" |
nothing calls this directly
no test coverage detected