Updated mlflow and scikit-learn (#3130)

* Updated mlflow and scikit-learn

* Changed to use latest sklearn 1.0 curated environment

* Update Pytorch environment

Author: jeff-shepherd

cli/assets/environment/docker-contexts/python-and-pip/requirements.txt

@@ -2,7 +2,7 @@
 numpy==1.22
 scipy==1.7.1
 pandas==1.3.0
-scikit-learn==0.24.2
+scikit-learn==1.2.2
 adlfs==2021.9.1
 fsspec==2021.8.1
 
@@ -11,8 +11,8 @@ xgboost==1.4.2
 lightgbm==3.2.1
 
 # tracking
-mlflow==1.20.2
-azureml-mlflow==1.34.0
+mlflow==2.2.1
+azureml-mlflow==1.55.0
 
 # visualization
 matplotlib==3.4.3

tutorials/e2e-distributed-pytorch-image/e2e-object-classification-distributed-pytorch.ipynb

@@ -241,7 +241,7 @@
     "        ),\n",
     "    },\n",
     "    # we're using a curated environment\n",
-    "    environment=\"AzureML-sklearn-1.0-ubuntu20.04-py38-cpu:1\",\n",
+    "    environment=\"AzureML-sklearn-1.0-ubuntu20.04-py38-cpu@latest\",\n",
     "    compute=\"cpu-cluster\"\n",
     "    if (cpu_cluster)\n",
     "    else None,  # No compute needs to be passed to use serverless\n",
@@ -325,7 +325,7 @@
     "        ),\n",
     "    },\n",
     "    # we're using a curated environment\n",
-    "    environment=\"AzureML-sklearn-1.0-ubuntu20.04-py38-cpu:1\",\n",
+    "    environment=\"AzureML-sklearn-1.0-ubuntu20.04-py38-cpu@latest\",\n",
     "    compute=\"cpu-cluster-lg\",\n",
     ")\n",
     "\n",
@@ -412,7 +412,7 @@
     "        # \"register_model_as\": \"places_dev\",\n",
     "        \"enable_profiling\": False,\n",
     "    },\n",
-    "    environment=\"AzureML-pytorch-1.10-ubuntu18.04-py38-cuda11-gpu@latest\",\n",
+    "    environment=\"AzureML-acpt-pytorch-1.13-cuda11.7@latest\",\n",
     "    compute=\"gpu-cluster\"\n",
     "    if (gpu_cluster)\n",
     "    else None,  # No compute needs to be passed to use serverless\n",

tutorials/e2e-distributed-pytorch-image/requirements.txt

@@ -4,7 +4,7 @@ torch==1.10.2
 
 # for metrics reporting/plotting
 azureml-mlflow==1.40.0
-mlflow==1.23.1
+mlflow==2.2.1
 matplotlib==3.5.1
 
 # to run the notebooks locally