run_sm.py

""" Python script that triggers sagemaker flow"""

import sys
import subprocess

def main():
    # Let's skip arg names
    repo = sys.argv[2]
    repo_name = repo.split('/')[-1]
    script_name = sys.argv[4]
    cmd = f'git clone {repo} && cd {repo_name} && sh {script_name}'
    # subprocess.call(cmd, shell=True)
    raise ValueError(cmd)

if __name__ == '__main__':
    main()