fix: parallel related samples failed in inference (#3261)

Azure · Jul 1, 2024 · d4c90ee · d4c90ee
1 parent 13fb4f2
commit d4c90ee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/cli/jobs/pipelines/iris-batch-prediction-using-parallel/environment/environment_parallel.yml b/cli/jobs/pipelines/iris-batch-prediction-using-parallel/environment/environment_parallel.yml
@@ -12,6 +12,5 @@ dependencies:
       - pandas
       - pillow
       - azureml-core
-      - scikit-learn~=0.20.0
+      - scikit-learn==0.20.3
       - cloudpickle==1.1.1
-      - tensorflow==2.1.4
diff --git a/sdk/python/jobs/pipelines/1g_pipeline_with_parallel_nodes/src/environment_parallel.yml b/sdk/python/jobs/pipelines/1g_pipeline_with_parallel_nodes/src/environment_parallel.yml
@@ -12,6 +12,5 @@ dependencies:
       - pandas
       - pillow
       - azureml-core
-      - scikit-learn~=0.20.0
+      - scikit-learn==0.20.3
       - cloudpickle==1.1.1
-      - tensorflow==2.1.4