fix: Update telmetry constants to include MLOPS as well

Author: Roja Reddy Sareddy

sagemaker-core/src/sagemaker/core/jumpstart/models.py

@@ -398,6 +398,7 @@ class CapabilityEnum(StrEnum):
     TRAINING = "TRAINING"
     FINE_TUNING = "FINE_TUNING"
     VALIDATION = "VALIDATION"
+    INCREMENTAL_TRAINING = "INCREMENTAL_TRAINING"
 
 
 class DemoNotebookModel(BaseConfig):

sagemaker-core/tests/unit/telemetry/test_telemetry_logging.py

@@ -155,7 +155,7 @@ def test_telemetry_emitter_decorator_success(
         )
 
         mock_send_telemetry_request.assert_called_once_with(
-            1, [1, 2], MOCK_SESSION, None, None, expected_extra_str
+            1, [11, 12], MOCK_SESSION, None, None, expected_extra_str
         )
 
     @pytest.mark.skipif(

sagemaker-mlops/tests/unit/workflow/test_pipeline.py

@@ -26,6 +26,7 @@ def mock_session():
     session.boto_session.client.return_value = Mock()
     session.sagemaker_client = Mock()
     session.local_mode = False
+    session.sagemaker_config = {}
     return session
 
 

sagemaker-mlops/tests/unit/workflow/test_pipeline_class.py

@@ -34,6 +34,7 @@ def mock_session():
     session.boto_session = Mock()
     session.boto_session.client = Mock(return_value=Mock())
     session.local_mode = False
+    session.sagemaker_config = {}
     session._append_sagemaker_config_tags = Mock(return_value=[])
     return session