AutoML Vision and NLP SDK pipeline fix (#3250)

* adding continue_on_step_failure False and force_rerun True for vision pipelines * adding continue_on_step_failure False and force_rerun True for vision pipelines * adding continue_on_step_failure False and force_rerun True for vision pipelines * update env for model registration custom component
Azure · Jun 20, 2024 · f80bb1c · f80bb1c
1 parent 9be775a
commit f80bb1c
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/...ification-multiclass-in-pipeline/automl-image-classification-multiclass-in-pipeline.ipynb b/...ification-multiclass-in-pipeline/automl-image-classification-multiclass-in-pipeline.ipynb
@@ -410,7 +410,7 @@
     "        command=\"python register.py \"\n",
     "        + \"--model_input_path ${{inputs.model_input_path}} \"\n",
     "        + \"--model_base_name ${{inputs.model_base_name}}\",\n",
-    "        environment=\"azureml-sklearn-1.1:30\",\n",
+    "        environment=\"azureml://registries/azureml/environments/sklearn-1.1/labels/latest\",\n",
     "    )\n",
     "    register_model = command_func(\n",
     "        model_input_path=image_classification_node.outputs.best_model\n",

diff --git a/...ification-multilabel-in-pipeline/automl-image-classification-multilabel-in-pipeline.ipynb b/...ification-multilabel-in-pipeline/automl-image-classification-multilabel-in-pipeline.ipynb
@@ -416,7 +416,7 @@
     "        command=\"python register.py \"\n",
     "        + \"--model_input_path ${{inputs.model_input_path}} \"\n",
     "        + \"--model_base_name ${{inputs.model_base_name}}\",\n",
-    "        environment=\"azureml-sklearn-1.1:30\",\n",
+    "        environment=\"azureml://registries/azureml/environments/sklearn-1.1/labels/latest\",\n",
     "    )\n",
     "    register_model = command_func(\n",
     "        model_input_path=image_classification_multilabel_node.outputs.best_model\n",

diff --git a/...ge-instance-segmentation-in-pipeline/automl-image-instance-segmentation-in-pipeline.ipynb b/...ge-instance-segmentation-in-pipeline/automl-image-instance-segmentation-in-pipeline.ipynb
@@ -373,7 +373,7 @@
     "        command=\"python register.py \"\n",
     "        + \"--model_input_path ${{inputs.model_input_path}} \"\n",
     "        + \"--model_base_name ${{inputs.model_base_name}}\",\n",
-    "        environment=\"azureml-sklearn-1.1:30\",\n",
+    "        environment=\"azureml://registries/azureml/environments/sklearn-1.1/labels/latest\",\n",
     "    )\n",
     "    register_model = command_func(\n",
     "        model_input_path=image_instance_segmentation_node.outputs.best_model\n",

diff --git a/...automl-image-object-detection-in-pipeline/automl-image-object-detection-in-pipeline.ipynb b/...automl-image-object-detection-in-pipeline/automl-image-object-detection-in-pipeline.ipynb
@@ -428,7 +428,7 @@
     "        command=\"python register.py \"\n",
     "        + \"--model_input_path ${{inputs.model_input_path}} \"\n",
     "        + \"--model_base_name ${{inputs.model_base_name}}\",\n",
-    "        environment=\"azureml-sklearn-1.1:30\",\n",
+    "        environment=\"azureml://registries/azureml/environments/sklearn-1.1/labels/latest\",\n",
     "    )\n",
     "    register_model = command_func(\n",
     "        model_input_path=image_object_detection_node.outputs.best_model\n",

diff --git a/...eline/automl-text-classification-in-pipeline/automl-text-classification-in-pipeline.ipynb b/...eline/automl-text-classification-in-pipeline/automl-text-classification-in-pipeline.ipynb
@@ -188,7 +188,7 @@
     "        command=\"python register.py \"\n",
     "        + \"--model_input_path ${{inputs.model_input_path}} \"\n",
     "        + \"--model_base_name ${{inputs.model_base_name}}\",\n",
-    "        environment=\"azureml-sklearn-1.1:30\",\n",
+    "        environment=\"azureml://registries/azureml/environments/sklearn-1.1/labels/latest\",\n",
     "    )\n",
     "    register_model = command_func(\n",
     "        model_input_path=text_classification_node.outputs.best_model\n",

diff --git a/...sification-multilabel-in-pipeline/automl-text-classification-multilabel-in-pipeline.ipynb b/...sification-multilabel-in-pipeline/automl-text-classification-multilabel-in-pipeline.ipynb
@@ -187,7 +187,7 @@
     "        command=\"python register.py \"\n",
     "        + \"--model_input_path ${{inputs.model_input_path}} \"\n",
     "        + \"--model_base_name ${{inputs.model_base_name}}\",\n",
-    "        environment=\"azureml-sklearn-1.1:30\",\n",
+    "        environment=\"azureml://registries/azureml/environments/sklearn-1.1/labels/latest\",\n",
     "    )\n",
     "    register_model = command_func(\n",
     "        model_input_path=text_classification_multilabel_node.outputs.best_model\n",

diff --git a/...entity-recognition-in-pipeline/automl-text-ner-named-entity-recognition-in-pipeline.ipynb b/...entity-recognition-in-pipeline/automl-text-ner-named-entity-recognition-in-pipeline.ipynb
@@ -183,7 +183,7 @@
     "        command=\"python register.py \"\n",
     "        + \"--model_input_path ${{inputs.model_input_path}} \"\n",
     "        + \"--model_base_name ${{inputs.model_base_name}}\",\n",
-    "        environment=\"azureml-sklearn-1.1:30\",\n",
+    "        environment=\"azureml://registries/azureml/environments/sklearn-1.1/labels/latest\",\n",
     "    )\n",
     "    register_model = command_func(model_input_path=text_ner_node.outputs.best_model)\n",
     "\n",