diff --git a/magemaker/runner.py b/magemaker/runner.py index 6de523e..53d7998 100644 --- a/magemaker/runner.py +++ b/magemaker/runner.py @@ -161,6 +161,13 @@ def runner(): help="Show magemaker version and exit" ) + parser.add_argument( + '--query', + action='query', + help="path to YAML query configuration file", + type=str + ) + parser.add_argument( "--hf", help="Deploy a Hugging Face Model.",