Fix: Add validation to bedrock reward models

Author: Roja Reddy Sareddy

sagemaker-train/src/sagemaker/train/constants.py

@@ -43,7 +43,7 @@
 HUB_NAME = "SageMakerPublicHub"
 
 # Allowed reward model IDs for RLAIF trainer
-ALLOWED_REWARD_MODEL_IDS = [
+_ALLOWED_REWARD_MODEL_IDS = [
     "openai.gpt-oss-120b-1:0",
     "openai.gpt-oss-20b-1:0", 
     "qwen.qwen3-32b-v1:0",

sagemaker-train/src/sagemaker/train/rlaif_trainer.py

@@ -26,7 +26,7 @@
 )
 from sagemaker.core.telemetry.telemetry_logging import _telemetry_emitter
 from sagemaker.core.telemetry.constants import Feature
-from sagemaker.train.constants import HUB_NAME, ALLOWED_REWARD_MODEL_IDS
+from sagemaker.train.constants import HUB_NAME, _ALLOWED_REWARD_MODEL_IDS
 
 logger = logging.getLogger(__name__)
 
@@ -170,10 +170,10 @@ def _validate_reward_model_id(self, reward_model_id):
         if not reward_model_id:
             return None
 
-        if reward_model_id not in ALLOWED_REWARD_MODEL_IDS:
+        if reward_model_id not in _ALLOWED_REWARD_MODEL_IDS:
             raise ValueError(
                 f"Invalid reward_model_id '{reward_model_id}'. "
-                f"Available models are: {ALLOWED_REWARD_MODEL_IDS}"
+                f"Available models are: {_ALLOWED_REWARD_MODEL_IDS}"
             )
         return reward_model_id