utils/third_party/nni/recoverable.py

# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.

import os

class Recoverable:

    def load_checkpoint(self):
        pass

    def save_checkpoint(self):
        pass

    def get_checkpoint_path(self):
        ckp_path = os.getenv('NNI_CHECKPOINT_DIRECTORY')
        if ckp_path is not None and os.path.isdir(ckp_path):
            return ckp_path
        return None