Skip to content

Commit 7441bc4

Browse files
Pin numpy dependency (Azure#3441)
* Pin numpy dependency * update
1 parent 47af267 commit 7441bc4

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

tutorials/get-started-notebooks/quickstart.ipynb

+22
Original file line numberDiff line numberDiff line change
@@ -252,10 +252,32 @@
252252
" ##########################\n",
253253
" # Registering the model to the workspace\n",
254254
" print(\"Registering the model via MLFlow\")\n",
255+
"\n",
256+
" # pin numpy\n",
257+
" conda_env = {\n",
258+
" 'name': 'mlflow-env',\n",
259+
" 'channels': ['conda-forge'],\n",
260+
" 'dependencies': [\n",
261+
" 'python=3.10.15',\n",
262+
" 'pip<=21.3.1',\n",
263+
" {\n",
264+
" 'pip': [\n",
265+
" 'mlflow==2.17.0',\n",
266+
" 'cloudpickle==2.2.1',\n",
267+
" 'pandas==1.5.3',\n",
268+
" 'psutil==5.8.0',\n",
269+
" 'scikit-learn==1.5.2',\n",
270+
" 'numpy==1.26.4',\n",
271+
" ]\n",
272+
" }\n",
273+
" ],\n",
274+
" }\n",
275+
"\n",
255276
" mlflow.sklearn.log_model(\n",
256277
" sk_model=clf,\n",
257278
" registered_model_name=args.registered_model_name,\n",
258279
" artifact_path=args.registered_model_name,\n",
280+
" conda_env=conda_env,\n",
259281
" )\n",
260282
"\n",
261283
" # Saving the model to a file\n",

0 commit comments

Comments
 (0)