[console_scripts]
serve = sagemaker_containers.cli.serve:main
train = sagemaker_containers.cli.train:main

