Updated tests to use created environment in the job

Azure · Dec 18, 2024 · 7377384 · 7377384
1 parent 2afe872
commit 7377384
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 12 deletions.
diff --git a/...rence/environments/minimal-ubuntu20.04-py39-cpu-inference/tests/minimal_inference_test.py b/...rence/environments/minimal-ubuntu20.04-py39-cpu-inference/tests/minimal_inference_test.py
@@ -32,10 +32,10 @@ def test_minimal_cpu_inference():
 
     env_docker_context = Environment(
         build=BuildContext(path=this_dir / BUILD_CONTEXT),
-        name="minimal_cpu_inference",
+        name=env_name,
         description="minimal 20.04 py39 cpu inference environment created from a Docker context.",
     )
-    ml_client.environments.create_or_update(env_docker_context)
+    returned_env = ml_client.environments.create_or_update(env_docker_context)
 
     # create the command
     job = command(
@@ -44,7 +44,7 @@ def test_minimal_cpu_inference():
         inputs=dict(
             score="valid_score.py",
         ),
-        environment=f"{env_name}@latest",
+        environment=returned_env,
         compute=os.environ.get("cpu_cluster"),
         display_name="minimal-cpu-inference-example",
         description="A test run of the minimal 20.04 py39 cpu inference curated environment",

diff --git a/...ence/environments/minimal-ubuntu22.04-py311-cpu-inference/tests/minimal_inference_test.py b/...ence/environments/minimal-ubuntu22.04-py311-cpu-inference/tests/minimal_inference_test.py
@@ -32,10 +32,10 @@ def test_minimal_cpu_inference():
 
     env_docker_context = Environment(
         build=BuildContext(path=this_dir / BUILD_CONTEXT),
-        name="minimal_cpu_inference",
+        name=env_name,
         description="minimal 22.04 py311 cpu inference environment created from a Docker context.",
     )
-    ml_client.environments.create_or_update(env_docker_context)
+    returned_env = ml_client.environments.create_or_update(env_docker_context)
 
     # create the command
     job = command(
@@ -44,7 +44,7 @@ def test_minimal_cpu_inference():
         inputs=dict(
             score="valid_score.py",
         ),
-        environment=f"{env_name}@latest",
+        environment=returned_env,
         compute=os.environ.get("cpu_cluster"),
         display_name="minimal-cpu-inference-example",
         description="A test run of the minimal 22.04 py311 cpu inference curated environment",

diff --git a/...erence/environments/mlflow-ubuntu20.04-py39-cpu-inference/context/conda_dependencies.yaml b/...erence/environments/mlflow-ubuntu20.04-py39-cpu-inference/context/conda_dependencies.yaml
@@ -8,6 +8,4 @@ dependencies:
 - pip:
   - azureml-inference-server-http=={{latest-pypi-version}}
   - azureml-ai-monitoring=={{latest-pypi-version}}
-  - numpy
-  - mlflow
-  - azureml-contrib-services
+  - mlflow
diff --git a/assets/inference/environments/mlflow-ubuntu20.04-py39-cpu-inference/tests/mlflow_test.py b/assets/inference/environments/mlflow-ubuntu20.04-py39-cpu-inference/tests/mlflow_test.py
@@ -32,10 +32,10 @@ def test_mlflow_cpu_inference():
 
     env_docker_context = Environment(
         build=BuildContext(path=this_dir / BUILD_CONTEXT),
-        name="mlflow_py39_inference",
+        name=env_name,
         description="mlflow 20.04 py39 cpu inference environment created from a Docker context.",
     )
-    ml_client.environments.create_or_update(env_docker_context)
+    returned_env = ml_client.environments.create_or_update(env_docker_context)
 
     # create the command
     job = command(
@@ -47,7 +47,7 @@ def test_mlflow_cpu_inference():
             score_input="sample_2_0_input.txt",
             model_dir="mlflow_2_0_model_folder"
         ),
-        environment=f"{env_name}@latest",
+        environment=returned_env,
         compute=os.environ.get("cpu_cluster"),
         display_name="mlflow-py39-inference-example",
         description="A test run of the mlflow 20.04 py39 cpu inference curated environment",