(self)
| 488 | self.num_local_samples += 1 |
| 489 | |
| 490 | def close(self): |
| 491 | if self._closed: |
| 492 | self._raise_if_failed() |
| 493 | return |
| 494 | if self._exception is None: |
| 495 | self._put(self.sentinel) |
| 496 | self.thread.join() |
| 497 | self._closed = True |
| 498 | self._raise_if_failed() |
| 499 | assert self.writer.num_local_samples == self.num_local_samples, ( |
| 500 | "Async target cache writer lost samples: " |
| 501 | f"{self.writer.num_local_samples} != {self.num_local_samples}" |
| 502 | ) |
| 503 | |
| 504 | |
| 505 | def load_local_cache_write_summary(rank_dir: str): |
no test coverage detected