| """ Python script that triggers sagemaker flow""" | |
| import sys | |
| import subprocess | |
| def main(): | |
| # Let's skip arg names | |
| script_name = sys.argv[2] | |
| repo = sys.argv[4] | |
| repo_name = repo.split('/')[-1] | |
| cmd = f'git clone {repo} && cd {repo_name} && sh {script_name}' | |
| output = subprocess.run(cmd.split(), stdout=subprocess.PIPE) | |
| print(output.stdout.decode()) | |
| if __name__ == '__main__': | |
| main() | |