fix: load model path and function

Azure · Jun 30, 2023 · 7f83c51 · 7f83c51
1 parent 3bda898
commit 7f83c51
Show file tree

Hide file tree

Showing 12 changed files with 4 additions and 52 deletions.
diff --git a/...points/batch/deploy-models/heart-classifier-mlflow/deployment-custom/code/batch_driver.py b/...points/batch/deploy-models/heart-classifier-mlflow/deployment-custom/code/batch_driver.py
@@ -19,7 +19,7 @@ def init():
     model_path = glob.glob(os.environ["AZUREML_MODEL_DIR"] + "/*/")[0]
 
     # Load the model, it's input types and output names
-    model = mlflow.pyfunc.load(model_path)
+    model = mlflow.pyfunc.load_model(model_path)
     if model.metadata and model.metadata.signature:
         if model.metadata.signature.inputs:
             model_input_types = dict(

diff --git a/...atch/deploy-models/heart-classifier-mlflow/deployment-custom/code/batch_driver_parquet.py b/...atch/deploy-models/heart-classifier-mlflow/deployment-custom/code/batch_driver_parquet.py
@@ -22,7 +22,7 @@ def init():
     output_path = os.environ["AZUREML_BI_OUTPUT_PATH"]
 
     # Load the model, it's input types and output names
-    model = mlflow.pyfunc.load(model_path)
+    model = mlflow.pyfunc.load_model(model_path)
     if model.metadata and model.metadata.signature:
         if model.metadata.signature.inputs:
             model_input_types = dict(

diff --git a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/MLmodel b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/MLmodel
diff --git a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/conda.yaml b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/conda.yaml
diff --git a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/model.pkl b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/model.pkl
diff --git a/...ints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/requirements.txt b/...ints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/requirements.txt
diff --git a/...er-mlflow/deployment-custom/model/MLmodel → ...els/heart-classifier-mlflow/model/MLmodel b/...er-mlflow/deployment-custom/model/MLmodel → ...els/heart-classifier-mlflow/model/MLmodel
diff --git a/...mlflow/deployment-custom/model/conda.yaml → .../heart-classifier-mlflow/model/conda.yaml b/...mlflow/deployment-custom/model/conda.yaml → .../heart-classifier-mlflow/model/conda.yaml
diff --git a/...-mlflow/deployment-custom/model/model.pkl → ...s/heart-classifier-mlflow/model/model.pkl b/...-mlflow/deployment-custom/model/model.pkl → ...s/heart-classifier-mlflow/model/model.pkl
diff --git a/.../deployment-custom/model/requirements.txt → ...-classifier-mlflow/model/requirements.txt b/.../deployment-custom/model/requirements.txt → ...-classifier-mlflow/model/requirements.txt
diff --git a/sdk/python/endpoints/batch/deploy-models/heart-classifier-mlflow/code/batch_driver.py b/sdk/python/endpoints/batch/deploy-models/heart-classifier-mlflow/code/batch_driver.py
@@ -19,7 +19,7 @@ def init():
     model_path = glob.glob(os.environ["AZUREML_MODEL_DIR"] + "/*/")[0]
 
     # Load the model, it's input types and output names
-    model = mlflow.pyfunc.load(model_path)
+    model = mlflow.pyfunc.load_model(model_path)
     if model.metadata and model.metadata.signature:
         if model.metadata.signature.inputs:
             model_input_types = dict(

diff --git a/...python/endpoints/batch/deploy-models/heart-classifier-mlflow/code/batch_driver_parquet.py b/...python/endpoints/batch/deploy-models/heart-classifier-mlflow/code/batch_driver_parquet.py
@@ -22,7 +22,7 @@ def init():
     output_path = os.environ["AZUREML_BI_OUTPUT_PATH"]
 
     # Load the model, it's input types and output names
-    model = mlflow.pyfunc.load(model_path)
+    model = mlflow.pyfunc.load_model(model_path)
     if model.metadata and model.metadata.signature:
         if model.metadata.signature.inputs:
             model_input_types = dict(