From ba90f6b0cffffb78831fbb8800cd58012fd9f575 Mon Sep 17 00:00:00 2001
From: aviruthen <91846056+aviruthen@users.noreply.github.com>
Date: Tue, 7 Apr 2026 19:47:46 -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 6cc201888835f975d275469d0265992cda052318 Mon Sep 17 00:00:00 2001
From: aviruthen <91846056+aviruthen@users.noreply.github.com>
Date: Tue, 7 Apr 2026 19:51:34 -0400
Subject: [PATCH 2/2] fix: address review comments (iteration #2)

---
 src/sagemaker/workflow/pipeline.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/sagemaker/workflow/pipeline.py b/src/sagemaker/workflow/pipeline.py
index c409b215f9..34b987c5d8 100644
--- a/src/sagemaker/workflow/pipeline.py
+++ b/src/sagemaker/workflow/pipeline.py
@@ -1,3 +1,4 @@
+# This is a test comment to verify the file can be modified.
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0