Archive model for Amazon SageMaker

archive.py

import tarfile
import os

root = "./endpoint"


def main():
    cwd = os.getcwd()
    os.chdir(root)
    files = [os.path.join(path, name) for path, _, files in os.walk(".") for name in files]
    with tarfile.open(os.path.join(cwd, "model.tar.gz"), "w:gz") as tar:
        for file in files:
            tar.add(file)


if __name__ == "__main__":
    main()