Added mlmonitoring to mlflow py39

assets/inference/environments/mlflow-ubuntu20.04-py39-cpu-inference/context/Dockerfile

@@ -13,8 +13,6 @@ ENV AML_APP_ROOT="/var/mlflow_resources"
 ENV AZUREML_ENTRY_SCRIPT="mlflow_score_script.py"
 
 USER root
-# Copying of mlmonitoring will add once testing is completed.
-# COPY mlmonitoring /var/mlflow_resources/mlmonitoring
 
 # We'll copy the HF scripts as well to enable better handling for v2 packaging. This will not require changes to the
 # packages installed in the image, as the expectation is that these will all be brought along with the model.

assets/inference/environments/mlflow-ubuntu20.04-py39-cpu-inference/context/conda_dependencies.yaml

@@ -6,4 +6,6 @@ dependencies:
 - python=3.9.13
 - pip
 - pip:
-  - azureml-inference-server-http=={{latest-pypi-version}}
+  - azureml-inference-server-http=={{latest-pypi-version}}
+  - azureml-ai-monitoring=={{latest-pypi-version}}
+  

assets/inference/environments/mlflow-ubuntu20.04-py39-cpu-inference/context/mlflow_score_script.py

@@ -17,7 +17,7 @@ from inference_schema.schema_decorators import input_schema, output_schema
 from mlflow.models import Model
 from mlflow.pyfunc import load_model
 from mlflow.pyfunc.scoring_server import _get_jsonable_obj
-from mlmonitoring import Collector
+from azureml.ai.monitoring import Collector
 from mlflow.types.utils import _infer_schema
 from mlflow.types.schema import Schema, ColSpec, DataType
 from mlflow.exceptions import MlflowException