Change example argument (mlflow#1521)

simonvanbernem · Jun 30, 2019 · 6b5c21d · 6b5c21d
1 parent e205be3
commit 6b5c21d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/models.rst b/docs/source/models.rst
@@ -429,7 +429,7 @@ evaluate test data.
     # Save the MLflow Model
     mlflow_pyfunc_model_path = "xgb_mlflow_pyfunc"
     mlflow.pyfunc.save_model(
-            dst_path=mlflow_pyfunc_model_path, python_model=XGBWrapper(), artifacts=artifacts,
+            path=mlflow_pyfunc_model_path, python_model=XGBWrapper(), artifacts=artifacts,
             conda_env=conda_env)
 
     # Load the model in `python_function` format