Function _checkpointer_config

axlearn/common/checkpointer_test.py:69–77

(
    checkpointer_cls: Type[BaseCheckpointer] = Checkpointer,
)

def _checkpointer_config(
    checkpointer_cls: Type[BaseCheckpointer] = Checkpointer,
) -> BaseCheckpointer.Config:
    # TODO(markblee): Use context manager instead of mkdtemp.
    return checkpointer_cls.default_config().set(
        name="test",
        dir=tempfile.mkdtemp(),
        keep_last_n=1,
    )


# pylint: disable=protected-access