From 3736f3c811d0715a1ba090bc557cbabc2297432d Mon Sep 17 00:00:00 2001
From: aviruthen <91846056+aviruthen@users.noreply.github.com>
Date: Wed, 3 Jun 2026 14:54:09 -0700
Subject: [PATCH] fix: rename lr_warmup_ratio to lr_warmup_steps_ratio to match
 Hub recipe schema

---
 src/sagemaker/hyperpod/cli/recipe_param_order.py        | 2 +-
 test/integration_tests/init/test_recipe_job_creation.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/sagemaker/hyperpod/cli/recipe_param_order.py b/src/sagemaker/hyperpod/cli/recipe_param_order.py
index 0eec314a..aed682ed 100644
--- a/src/sagemaker/hyperpod/cli/recipe_param_order.py
+++ b/src/sagemaker/hyperpod/cli/recipe_param_order.py
@@ -49,7 +49,7 @@
     ("max_steps",               "Core Hyperparameters"),
 
     # Advanced Hyperparameters (includes technique-specific params)
-    ("lr_warmup_ratio",         "Advanced Hyperparameters"),
+    ("lr_warmup_steps_ratio",   "Advanced Hyperparameters"),
     ("max_context_length",      "Advanced Hyperparameters"),
     ("max_prompt_length",       "Advanced Hyperparameters"),
     ("max_length",              "Advanced Hyperparameters"),
diff --git a/test/integration_tests/init/test_recipe_job_creation.py b/test/integration_tests/init/test_recipe_job_creation.py
index 5e6743be..a0f89b27 100644
--- a/test/integration_tests/init/test_recipe_job_creation.py
+++ b/test/integration_tests/init/test_recipe_job_creation.py
@@ -97,7 +97,7 @@ def test_configure_recipe_job(runner, job_name, test_directory):
         "--data-path", "/data/recipes-data/sft/zc_train_256.jsonl",
         "--global-batch-size", "8",
         "--learning-rate", "0.0001",
-        "--lr-warmup-ratio", "0.1",
+        "--lr-warmup-steps-ratio", "0.1",
         "--max-epochs", "20", 
         "--output-path", "/data/output/qwen3-sft",
         "--results-directory", "/data/results/qwen3-sft",