Using aws ssm to execute python script that takes an argument

Question

I have a python script on an ec2 instance that I can run by sshing into it and executing as such

python /root/path/to/file/main.py local

here