Fix jumpstart integ test (#5351)

zhaoqizqwang · web-flow · commit c05f0e7406c9 · 2025-12-02T20:45:36.000-08:00
diff --git 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):
