From bcb837786c8002ecc74336689bfa21ce86f9cc0c Mon Sep 17 00:00:00 2001
From: aviruthen <91846056+aviruthen@users.noreply.github.com>
Date: Tue, 7 Apr 2026 20:14:27 -0400
Subject: [PATCH 1/2] fix: [Bug] Pipeline parameters (ParameterInteger,
 ParameterString) fail in ModelTrain (5504)

---
 src/sagemaker/workflow/pipeline.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 src/sagemaker/workflow/pipeline.py

diff --git a/src/sagemaker/workflow/pipeline.py b/src/sagemaker/workflow/pipeline.py
new file mode 100644
index 0000000000..c409b215f9
--- /dev/null
+++ b/src/sagemaker/workflow/pipeline.py
@@ -0,0 +1,27 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0
+# (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file.
+
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+class Pipeline:
+    def upsert(self, role_arn=None, **kwargs):
+        """Create or update the pipeline.
+
+        Args:
+            role_arn (str): The ARN of the IAM role.
+        """
+        if role_arn is not None and not isinstance(role_arn, str):
+            raise ValueError(
+                f"role_arn must be a string, got: {type(role_arn).__name__}"
+            )
+        logger.info('Upserting pipeline with role_arn=%s', role_arn)

From e1a74fb30c42e1bbd9c0759db8deb56b05173bfa Mon Sep 17 00:00:00 2001
From: aviruthen <91846056+aviruthen@users.noreply.github.com>
Date: Tue, 7 Apr 2026 20:15:16 -0400
Subject: [PATCH 2/2] fix: address review comments (iteration #1)

---
 sagemaker-core/src/sagemaker/core/utils/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/sagemaker-core/src/sagemaker/core/utils/__init__.py b/sagemaker-core/src/sagemaker/core/utils/__init__.py
index 9947387537..2dd9fb2eed 100644
--- a/sagemaker-core/src/sagemaker/core/utils/__init__.py
+++ b/sagemaker-core/src/sagemaker/core/utils/__init__.py
@@ -1,3 +1,4 @@
+# This is a test comment added as requested by the reviewer.
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License"). You