check that batchrun ins't a pipeline job (Azure#30840)

LazaUK · Jul 6, 2023 · 8b753de · 8b753de
1 parent c86ca6a
commit 8b753de
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/constants/_job/pipeline.py b/sdk/ml/azure-ai-ml/azure/ai/ml/constants/_job/pipeline.py
@@ -16,6 +16,7 @@ class PipelineConstants:
     REUSED_FLAG_FIELD = "azureml.isreused"
     REUSED_FLAG_TRUE = "true"
     REUSED_JOB_ID = "azureml.reusedrunid"
+    PIPELINE_JOB_TYPE = "azureml.pipelinejob"
 
 
 class ValidationErrorCode:

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_job_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_job_operations.py
@@ -714,7 +714,10 @@ def download(
                 error_category=ErrorCategory.USER_ERROR,
             )
 
-        is_batch_job = job_details.tags.get("azureml.batchrun", None) == "true"
+        is_batch_job = (
+            job_details.tags.get("azureml.batchrun", None) == "true"
+            and job_details.tags.get("azureml.jobtype", None) != PipelineConstants.PIPELINE_JOB_TYPE
+        )
         outputs = {}
         download_path = Path(download_path)
         artifact_directory_name = "artifacts"