Pin numpy dependency (#3441)

* Pin numpy dependency * update
Azure · Nov 8, 2024 · 7441bc4 · 7441bc4
1 parent 47af267
commit 7441bc4
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/tutorials/get-started-notebooks/quickstart.ipynb b/tutorials/get-started-notebooks/quickstart.ipynb
@@ -252,10 +252,32 @@
     "    ##########################\n",
     "    # Registering the model to the workspace\n",
     "    print(\"Registering the model via MLFlow\")\n",
+    "\n",
+    "    # pin numpy\n",
+    "    conda_env = {\n",
+    "        'name': 'mlflow-env',\n",
+    "        'channels': ['conda-forge'],\n",
+    "        'dependencies': [\n",
+    "            'python=3.10.15',\n",
+    "            'pip<=21.3.1',\n",
+    "            {\n",
+    "                'pip': [\n",
+    "                    'mlflow==2.17.0',\n",
+    "                    'cloudpickle==2.2.1',\n",
+    "                    'pandas==1.5.3',\n",
+    "                    'psutil==5.8.0',\n",
+    "                    'scikit-learn==1.5.2',\n",
+    "                    'numpy==1.26.4',\n",
+    "                ]\n",
+    "            }\n",
+    "        ],\n",
+    "    }\n",
+    "\n",
     "    mlflow.sklearn.log_model(\n",
     "        sk_model=clf,\n",
     "        registered_model_name=args.registered_model_name,\n",
     "        artifact_path=args.registered_model_name,\n",
+    "        conda_env=conda_env,\n",
     "    )\n",
     "\n",
     "    # Saving the model to a file\n",