| """ 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() | |