fix: trn1 instance family does not support volume size

Author: evakravi

src/sagemaker/utils.py

@@ -1462,6 +1462,7 @@ def volume_size_supported(instance_type: str) -> bool:
             and not family.startswith("g5")
             and not family.startswith("g6")
             and not family.startswith("p5")
+            and not family.startswith("trn1")
         )
     except Exception as e:
         raise ValueError(f"Failed to parse instance type '{instance_type}': {str(e)}")

tests/unit/test_utils.py

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