Fixing automl image segmentation test and increasing the gate timeout (#34381)

* increase timeout

* use mmd 3.x model name format

* parameterizing the timeout

Author: nvijayrania

sdk/ml/automl.tests.yml

@@ -4,6 +4,7 @@ extends:
   template: ./tests.yml
   parameters:
     TestMarkArgument: "automl_test"
+    TestTimeoutInMinutes: 480
     CloudConfig:
       Public:
         SubscriptionConfiguration: $(python-ml-automl-sub-scope)

sdk/ml/azure-ai-ml/tests/automl_job/e2etests/test_automl_image_segmentation.py

@@ -98,7 +98,7 @@ def test_image_segmentation_run(
             image_instance_segmentation_job_sweep.extend_search_space(
                 [
                     SearchSpace(
-                        model_name=Choice(["mask_rcnn_swin-s-p4-w7_fpn_fp16_ms-crop-3x_coco"]),
+                        model_name=Choice(["mask-rcnn_swin-t-p4-w7_fpn_amp-ms-crop-3x_coco"]),
                         number_of_epochs=Choice([1]),
                         gradient_accumulation_step=Choice([1]),
                         learning_rate=Choice([0.005]),
@@ -118,7 +118,7 @@ def test_image_segmentation_run(
             )
             image_instance_segmentation_job_individual = copy.deepcopy(image_instance_segmentation_job)
             image_instance_segmentation_job_individual.set_training_parameters(
-                model_name="mask_rcnn_swin-s-p4-w7_fpn_fp16_ms-crop-3x_coco", number_of_epochs=1
+                model_name="mask-rcnn_swin-t-p4-w7_fpn_amp-ms-crop-3x_coco", number_of_epochs=1
             )
             image_instance_segmentation_job_reuse = copy.deepcopy(image_instance_segmentation_job_individual)
         else:

sdk/ml/tests.yml

@@ -10,13 +10,17 @@ parameters:
         Location: eastus2euap
         MatrixReplace:
           - TestSamples=.*/true
+  - name: TestTimeoutInMinutes
+    type: number
+    default: 270
+  
 stages:
   - template: ../../eng/pipelines/templates/stages/archetype-sdk-tests.yml
     parameters:
       ServiceDirectory: ml
       MatrixReplace:
         - TestSamples=.*/true
-      TestTimeoutInMinutes: 270
+      TestTimeoutInMinutes: ${{ parameters.TestTimeoutInMinutes }}
       TestMarkArgument: ${{ parameters.TestMarkArgument }}
       CloudConfig: ${{ parameters.CloudConfig }}
       EnvVars: