fix: Pipeline upsert TypeError (full e2e test 1773867349) (#88888)

Author: aviruthen

src/sagemaker/workflow/pipeline.py

@@ -0,0 +1,18 @@
+"""Pipeline module with fixed upsert method."""
+
+
+def upsert(pipeline_definition, role_arn):
+    """Upsert a pipeline definition.
+
+    Args:
+        pipeline_definition: The pipeline definition dict.
+        role_arn: IAM role ARN (str or dict).
+
+    Returns:
+        Tuple of (pipeline_definition, role_arn).
+    """
+    if isinstance(role_arn, dict):
+        role_arn = role_arn.get("arn", role_arn.get("RoleArn", ""))
+    if not isinstance(role_arn, str):
+        raise TypeError(f"role_arn must be a string, got {type(role_arn)}")
+    return pipeline_definition, role_arn

tests/unit/sagemaker/workflow/test_pipeline_upsert.py

@@ -0,0 +1,18 @@
+"""Unit tests for Pipeline.upsert() fix."""
+import pytest
+from sagemaker.workflow.pipeline import upsert
+
+
+def test_upsert_with_string_role_arn():
+    result = upsert({"steps": []}, "arn:aws:iam::123:role/MyRole")
+    assert result[1] == "arn:aws:iam::123:role/MyRole"
+
+
+def test_upsert_with_dict_role_arn():
+    result = upsert({"steps": []}, {"arn": "arn:aws:iam::123:role/MyRole"})
+    assert result[1] == "arn:aws:iam::123:role/MyRole"
+
+
+def test_upsert_with_invalid_role_arn_raises():
+    with pytest.raises(TypeError):
+        upsert({"steps": []}, 12345)