Method wait_until_finished

axlearn/common/checkpointer.py:330–332

Waits for async writes to finish.

(self)

        raise NotImplementedError(type(self))

    def wait_until_finished(self):
        """Waits for async writes to finish."""
        raise NotImplementedError(type(self))

    def restore_from_dir(
        self,