Environment for Paige Model

assets/training/model_management/environments/mlflow-model-inference/context/mlflow_score_script.py

@@ -163,6 +163,9 @@ def create_param_sample(model_signature_params):
 
 
 model_path = os.path.normpath(os.path.join(os.getenv("AZUREML_MODEL_DIR"), "mlflow_model_folder/"))
+if os.getenv("MLFLOW_MODEL_FOLDER"):
+    model_path = os.path.normpath(os.path.join(os.getenv("AZUREML_MODEL_DIR"), os.getenv("MLFLOW_MODEL_FOLDER")))
+
 
 # model loaded here using mlfow.models import Model so we have access to the model signature
 model = Model.load(model_path)

assets/training/model_management/environments/mlflow-model-inference/context/runit/gunicorn/run

@@ -68,7 +68,13 @@ format_print "Dynamic Python Package Installation"
 format_print "###############################################"
 format_print ""
 
-export AZUREML_EXTRA_CONDA_YAML_ABS_PATH="${AZUREML_MODEL_DIR}/mlflow_model_folder/conda.yaml"
+ENV_MLFLOW_MODEL_FOLDER="mlflow_model_folder"
+if [[ -n "$MLFLOW_MODEL_FOLDER" ]]; then
+    ENV_MLFLOW_MODEL_FOLDER="${MLFLOW_MODEL_FOLDER}"
+fi
+export AZUREML_EXTRA_CONDA_YAML_ABS_PATH="${AZUREML_MODEL_DIR}/${ENV_MLFLOW_MODEL_FOLDER}/conda.yaml"
+echo "$AZUREML_EXTRA_CONDA_YAML_ABS_PATH"
+
 
 if [[ -n "$AZUREML_EXTRA_PYTHON_LIB_PATH" ]]; then
     # Pre-installed mounted dependencies, check for the variable and if the folder exists.

assets/training/model_management/tests/dev_conda_env.yaml

@@ -1,27 +1,28 @@
+name: model_mgmt
 channels:
-- conda-forge
+  - conda-forge
 dependencies:
-- python=3.8.0
-- pip<=23.1.2
-- pycocotools==2.0.4
-- pip:
-  ## src requirements
-  - mlflow~=2.10.2
-  - mlflow-skinny~=2.10.2
-  - azureml-mlflow~=1.55.0
-  - azure-ai-ml~=1.13.0
-  - azureml-core~=1.55.0
-  - azureml-automl-core~=1.55.0
-  - azureml-telemetry~=1.55.0
-  - azureml-evaluate-mlflow~=0.0.45
-  - langcodes
-  ## Test requirements
-  - mock
-  - pydash
-  - pytest
-  - pytest-cov
-  - pytest-forked
-  - pytest-mock
-  - pytest-timeout
-  - pytest-xdist
-name: model_mgmt
+  - python=3.10
+  - pip<=23.1.2
+  - pycocotools==2.0.4
+  - pip:
+      ## src requirements
+      - mlflow~=2.10.2
+      - mlflow-skinny~=2.10.2
+      - azureml-mlflow~=1.55.0
+      - azure-ai-ml~=1.13.0
+      - azureml-core~=1.55.0
+      - azureml-automl-core~=1.55.0
+      - azureml-telemetry~=1.55.0
+      - azureml-evaluate-mlflow~=0.0.45
+      - langcodes
+      - marshmallow==3.20.1
+      ## Test requirements
+      - mock
+      - pydash
+      - pytest
+      - pytest-cov
+      - pytest-forked
+      - pytest-mock
+      - pytest-timeout
+      - pytest-xdist