From 3e19a2d15579960e9f6dd4d7978675aa78635508 Mon Sep 17 00:00:00 2001
From: jzhaoqwa <52220743+zhaoqizqwang@users.noreply.github.com>
Date: Tue, 2 Dec 2025 19:56:08 -0800
Subject: [PATCH] Fix jumpstart integ test

---
 sagemaker-core/src/sagemaker/core/jumpstart/models.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/sagemaker-core/src/sagemaker/core/jumpstart/models.py b/sagemaker-core/src/sagemaker/core/jumpstart/models.py
index 65e940e58c..05021ed9d2 100644
--- a/sagemaker-core/src/sagemaker/core/jumpstart/models.py
+++ b/sagemaker-core/src/sagemaker/core/jumpstart/models.py
@@ -395,6 +395,9 @@ class CapabilityEnum(StrEnum):
     HYPERPOD_DEPLOYMENT = "HYPERPOD_DEPLOYMENT"
     SAGEMAKER_EVALUATE = "SAGEMAKER_EVALUATE"
     SAGEMAKER_MODEL_OPTIMIZE = "SAGEMAKER_MODEL_OPTIMIZE"
+    TRAINING = "TRAINING"
+    FINE_TUNING = "FINE_TUNING"
+    VALIDATION = "VALIDATION"
 
 
 class DemoNotebookModel(BaseConfig):