fix: skip intelligent volume_size allocation based on instance type if it is a pipeline parameter (#3972)

* fix: skip intelligent volume_size allocation based on instance type if it is a pipeline parameter

* fix: Change the order of the if statement check.

Author: jerrypeng7773

src/sagemaker/utils.py

@@ -1407,7 +1407,9 @@ def volume_size_supported(instance_type: str) -> bool:
     try:
 
         # local mode does not support volume size
-        if instance_type.startswith("local"):
+        # instance type given as pipeline parameter does not support volume size
+        # do not change the if statement order below.
+        if is_pipeline_variable(instance_type) or instance_type.startswith("local"):
             return False
 
         parts: List[str] = instance_type.split(".")

tests/unit/test_utils.py

@@ -1703,6 +1703,7 @@ def test_volume_size_not_supported(self):
             "g5.8xlarge",
             "local",
             "local_gpu",
+            ParameterString(name="InstanceType", default_value="ml.m4.xlarge"),
         ]
 
         for instance in instances_that_dont_support_volume_size: