From 51a1cf5fe14e9d00cb77c1ff17e1b5b9b320ac6f Mon Sep 17 00:00:00 2001
From: eff-kay <faizan10114@gmail.com>
Date: Mon, 24 Feb 2025 04:16:28 -0500
Subject: [PATCH] query param

---
 magemaker/runner.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/magemaker/runner.py b/magemaker/runner.py
index 6de523e..6d8cc2e 100644
--- a/magemaker/runner.py
+++ b/magemaker/runner.py
@@ -161,11 +161,19 @@ 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.",
         type=str
     )
+
     parser.add_argument(
         "--instance",
         help="EC2 instance type to deploy to.",