fix: Support primitive types for left value of ConditionSteps (#2886)

Co-authored-by: Payton Staub <[email protected]>
Co-authored-by: Navin Soni <[email protected]>

Author: 3 people

src/sagemaker/workflow/conditions.py

@@ -79,7 +79,7 @@ def to_request(self) -> RequestType:
         """Get the request structure for workflow service calls."""
         return {
             "Type": self.condition_type.value,
-            "LeftValue": self.left.expr,
+            "LeftValue": primitive_or_expr(self.left),
             "RightValue": primitive_or_expr(self.right),
         }
 

tests/unit/sagemaker/workflow/test_conditions.py

@@ -165,3 +165,12 @@ def test_condition_or():
             },
         ],
     }
+
+
+def test_left_and_right_primitives():
+    cond = ConditionEquals(left=2, right=1)
+    assert cond.to_request() == {
+        "Type": "Equals",
+        "LeftValue": 2,
+        "RightValue": 1,
+    }