def load_step()

in src/hyperpod_nemo_adapter/utils/callbacks/local_ckpt_io.py [0:0]

• 15 lines of code
• 5 McCabe index (conditional complexity)

    def load_step(fs, root):
        prev = None
        for d in _subdirs():
            file = fs.concat_path(root, d)
            path = fs.concat_path(file, ".local.metadata")
            meta = SageMakerLocalCheckpointIO.load_local_metadata(fs, path)
            step = meta.get("step", -1)
            if step < 0:
                return -1
            if not prev:
                prev = step
                continue
            if prev != step:
                return -1
        return step