diff --git a/sdk/python/jobs/parallel/2a_iris_batch_prediction/environment/environment_parallel.yml b/sdk/python/jobs/parallel/2a_iris_batch_prediction/environment/environment_parallel.yml
index 8cc8f51350..7478aa2330 100644
--- a/sdk/python/jobs/parallel/2a_iris_batch_prediction/environment/environment_parallel.yml
+++ b/sdk/python/jobs/parallel/2a_iris_batch_prediction/environment/environment_parallel.yml
@@ -2,15 +2,17 @@ name: prs-env
 channels:
   - conda-forge
 dependencies:
-  - python=3.7.6
+  - python=3.8
   - pip
   - pip:
-      - mlflow
+      - mlflow==1.30.1
       - mltable>=1.2.0
       - azureml-dataset-runtime[pandas,fuse]
       - azureml-telemetry
-      - pandas
+      - numpy==1.21.6
+      - scipy==1.7.3
+      - pandas==1.3.5
       - pillow
       - azureml-core
-      - scikit-learn~=0.20.0
-      - cloudpickle==1.1.1
+      - scikit-learn==0.23.2
+      - cloudpickle==1.6.0
diff --git a/sdk/python/jobs/parallel/2a_iris_batch_prediction/iris_batch_prediction.ipynb b/sdk/python/jobs/parallel/2a_iris_batch_prediction/iris_batch_prediction.ipynb
index 83a25575f6..b55402df30 100644
--- a/sdk/python/jobs/parallel/2a_iris_batch_prediction/iris_batch_prediction.ipynb
+++ b/sdk/python/jobs/parallel/2a_iris_batch_prediction/iris_batch_prediction.ipynb
@@ -170,7 +170,7 @@
     "        code=\"./script\",\n",
     "        entry_script=\"iris_prediction.py\",\n",
     "        environment=Environment(\n",
-    "            image=\"mcr.microsoft.com/azureml/openmpi4.1.0-ubuntu20.04\",\n",
+    "            image=\"mcr.microsoft.com/azureml/openmpi5.0-ubuntu24.04:latest\",\n",
     "            conda_file=\"./environment/environment_parallel.yml\",\n",
     "        ),\n",
     "        program_arguments=\"--model ${{inputs.score_model}} \"  # Passthrough input model folder path into script.\n",
diff --git a/sdk/python/mldesigner-0.1.0b23-py3-none-any.whl b/sdk/python/mldesigner-0.1.0b23-py3-none-any.whl
new file mode 100644
index 0000000000..0b6c9b4fb1
Binary files /dev/null and b/sdk/python/mldesigner-0.1.0b23-py3-none-any.whl differ
diff --git a/sdk/python/setup.sh b/sdk/python/setup.sh
index ec29a18244..ccf85de541 100644
--- a/sdk/python/setup.sh
+++ b/sdk/python/setup.sh
@@ -5,7 +5,7 @@
 # </az_ml_sdk_install>
 
 # <mldesigner_install>
-pip install mldesigner
+pip install mldesigner-0.1.0b23-py3-none-any.whl
 # </mldesigner_install>
 
 # <mltable_install>
@@ -23,4 +23,4 @@ pip install azure-ai-ml
 # protobuf==5.29.0 has IndentationError bug
 pip install "protobuf<=5.28.3"
 
-pip list
\ No newline at end of file
+pip list
diff --git a/sdk/python/setup_oai_v2.sh b/sdk/python/setup_oai_v2.sh
index 9078ae9d0d..a9b2e67be4 100644
--- a/sdk/python/setup_oai_v2.sh
+++ b/sdk/python/setup_oai_v2.sh
@@ -46,7 +46,7 @@ fi
 # </az_ml_sdk_install>
 
 # <mldesigner_install>
-pip install mldesigner
+pip install mldesigner-0.1.0b23-py3-none-any.whl
 # </mldesigner_install>
 
 # <mltable_install>
@@ -61,4 +61,4 @@ pip install azure-ai-ml
 # https://docsupport.blob.core.windows.net/ml-sample-submissions/1905732/azure_ai_ml-1.0.0-py3-none-any.whl
 # </az_ml_sdk_test_install>
 
-pip list
\ No newline at end of file
+pip list