Updated automl environments for SDK 1.59.0 (#3660)

* Updated automl environments for SDK 1.59.0
Azure · Dec 10, 2024 · 730981b · 730981b
1 parent 34f1ccb
commit 730981b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/assets/training/automl/environments/ai-ml-automl-dnn-text-gpu/context/Dockerfile b/assets/training/automl/environments/ai-ml-automl-dnn-text-gpu/context/Dockerfile
@@ -62,7 +62,7 @@ RUN HOROVOD_WITH_PYTORCH=1 pip install --no-cache-dir git+https://github.com/hor
 # by fixing dependencies in the base packages
 RUN pip list && \
     pip install pyarrow==14.0.2 \
-                'transformers[sentencepiece,torch]==4.36.2' \
+                'transformers[sentencepiece,torch]==4.37.2' \
                 aiohttp==3.10.2
 
 # end pip install

diff --git a/assets/training/automl/environments/ai-ml-automl/context/Dockerfile b/assets/training/automl/environments/ai-ml-automl/context/Dockerfile
@@ -47,7 +47,7 @@ RUN pip install \
                 azureml-dataset-runtime=={{latest-pypi-version}} \
                 azureml-defaults=={{latest-pypi-version}} \
                 # TODO: replace the hard coded above by a referenceto azureml-train-automl[tabular]
-                'mlflow-skinny==2.9.2' \
+                'mlflow-skinny==2.15.1' \
                 'xgboost==1.5.2' \
                 'prophet==1.1.4' \
                 'inference-schema' \