fix: default value removed if zero for integer param (#2409)

Author: melzareix

src/sagemaker/workflow/parameters.py

@@ -80,7 +80,7 @@ def to_request(self) -> RequestType:
             "Name": self.name,
             "Type": self.parameter_type.value,
         }
-        if self.default_value:
+        if self.default_value is not None:
             value["DefaultValue"] = self.default_value
         return value
 

tests/unit/sagemaker/workflow/test_parameters.py

@@ -37,6 +37,11 @@ def test_parameter_with_default():
     assert param.to_request() == {"Name": "MyFloat", "Type": "Float", "DefaultValue": 1.2}
 
 
+def test_parameter_with_default_value_zero():
+    param = ParameterInteger(name="MyInteger", default_value=0)
+    assert param.to_request() == {"Name": "MyInteger", "Type": "Integer", "DefaultValue": 0}
+
+
 def test_parameter_string_with_enum_values():
     param = ParameterString("MyString", enum_values=["a", "b"])
     assert param.to_request() == {"Name": "MyString", "Type": "String", "EnumValues": ["a", "b"]}