fix integ

Author: mollyheamazon

hyperpod-custom-inference-template/hyperpod_custom_inference_template/v1_0/model.py

@@ -251,8 +251,8 @@ def validate_model_source_config(self):
             if not self.s3_bucket_name or not self.s3_region:
                 raise ValueError("s3_bucket_name and s3_region are required when model_source_type is 's3'")
         elif self.model_source_type == "fsx":
-            if not self.fsx_dns_name or not self.fsx_file_system_id or not self.fsx_mount_name:
-                raise ValueError("fsx_dns_name, fsx_file_system_id and fsx_mount_name are required when model_source_type is 'fsx'")
+            if not self.fsx_file_system_id:
+                raise ValueError("fsx_file_system_id is required when model_source_type is 'fsx'")
         return self
 
     def to_domain(self) -> HPEndpoint:

test/integration_tests/inference/cli/test_cli_custom_fsx_inference.py

@@ -51,7 +51,6 @@ def test_custom_create(runner, custom_endpoint_name):
         "--model-source-type", "fsx",
         "--model-location", "hf-eqa",
         "--fsx-file-system-id", FSX_LOCATION,
-        "--s3-region", REGION,
         "--image-uri", "763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-inference:2.3.0-transformers4.48.0-cpu-py311-ubuntu22.04",
         "--container-port", "8080",
         "--model-volume-mount-name", "model-weights",