From 0bcba2f8a4357cd52ce8ceab10a4f26c0ca9299e Mon Sep 17 00:00:00 2001
From: Akshay Shah <akshayshah0803@gmail.com>
Date: Fri, 9 Oct 2026 20:56:52 -0500
Subject: [PATCH] fix(core): build a v3 ProcessingOutput in
 ModelMonitor._normalize_processing_output

ModelMonitor.run_baseline(output=<str>) normalizes the string through
_normalize_processing_output, which still built the output with the v2
ProcessingOutput(source=..., destination=...) signature. The v3
ProcessingOutput shape has no such fields (extra='forbid'), so every
run_baseline call with a string output raised a pydantic ValidationError.

Build it with output_name and an s3_output (ProcessingS3Output) instead,
keeping the v2 meaning: the string is the container path and the S3
destination is generated.
---
 .../core/model_monitor/model_monitoring.py    |  5 +++-
 .../tests/unit/test_model_monitoring.py       | 25 +++++++++++++++++++
 2 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/sagemaker-core/src/sagemaker/core/model_monitor/model_monitoring.py b/sagemaker-core/src/sagemaker/core/model_monitor/model_monitoring.py
index 27db390ac9..cc73f005ef 100644
--- a/sagemaker-core/src/sagemaker/core/model_monitor/model_monitoring.py
+++ b/sagemaker-core/src/sagemaker/core/model_monitor/model_monitoring.py
@@ -1470,7 +1470,10 @@ def _normalize_processing_output(self, output=None):
                 "output",
             )
             output = ProcessingOutput(
-                source=output, destination=s3_uri, output_name=_DEFAULT_OUTPUT_NAME
+                output_name=_DEFAULT_OUTPUT_NAME,
+                s3_output=ProcessingS3Output(
+                    s3_uri=s3_uri, local_path=output, s3_upload_mode="EndOfJob"
+                ),
             )
 
         return output
diff --git a/sagemaker-core/tests/unit/test_model_monitoring.py b/sagemaker-core/tests/unit/test_model_monitoring.py
index 9cb7dced11..5629b33112 100644
--- a/sagemaker-core/tests/unit/test_model_monitoring.py
+++ b/sagemaker-core/tests/unit/test_model_monitoring.py
@@ -363,6 +363,31 @@ def test_init_with_network_config(self, mock_session, test_role):
             )
             assert monitor.network_config is not None
 
+    def test_normalize_processing_output_from_local_path(self, mock_session, test_role):
+        """A string output is the container path; the S3 destination is generated."""
+        with (
+            patch(
+                "sagemaker.core.model_monitor.model_monitoring.resolve_value_from_config",
+                side_effect=lambda x, *args, **kwargs: x,
+            ),
+            patch(
+                "sagemaker.core.model_monitor.model_monitoring.resolve_class_attribute_from_config",
+                return_value=None,
+            ),
+        ):
+            monitor = ModelMonitor(
+                role=test_role, image_uri="test-image", sagemaker_session=mock_session
+            )
+        monitor.latest_baselining_job_name = "baseline-job"
+
+        output = monitor._normalize_processing_output(output="/opt/ml/processing/output")
+
+        assert isinstance(output, ProcessingOutput)
+        assert output.output_name == "monitoring_output"
+        assert output.s3_output.local_path == "/opt/ml/processing/output"
+        assert output.s3_output.s3_uri == "s3://test-bucket/test-prefix/baseline-job/output"
+        assert output.s3_output.s3_upload_mode == "EndOfJob"
+
     def test_generate_baselining_job_name_with_custom_name(self, mock_session, test_role):
         with (
             patch(