fix e2e tests

Azure · Jan 24, 2024 · 733d7af · 733d7af
1 parent 42d1f89
commit 733d7af
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 19 deletions.
diff --git a/assets/batch_score/components/driver/tests/e2e/aoai_endpoint_test.py b/assets/batch_score/components/driver/tests/e2e/aoai_endpoint_test.py
@@ -14,12 +14,7 @@
 cpu_compute_target = "cpu-cluster"
 source_dir = os.getcwd()
 gated_llm_pipeline_filepath = os.path.join(
-    source_dir,
-    "driver",
-    "tests",
-    "e2e",
-    "prs_pipeline_templates",
-    "base_llm.yml")
+    pytest.source_dir, "tests", "e2e", "prs_pipeline_templates", "base_llm.yml")
 
 RUN_NAME = "batch_score_aoai_endpoint_test"
 JOB_NAME = "gated_batch_score_llm"  # Should be equivalent to base_llm.yml's job name

diff --git a/assets/batch_score/components/driver/tests/e2e/conftest.py b/assets/batch_score/components/driver/tests/e2e/conftest.py
@@ -12,7 +12,6 @@
 from azure.ai.ml import MLClient, load_component
 from azure.identity import DefaultAzureCredential
 
-from src.batch_score.common.constants import METADATA_JSON_FILENAME
 from .util import _get_component_name, _set_and_get_component_name_ver, create_copy
 
 

diff --git a/assets/batch_score/components/driver/tests/e2e/mir_endpoint_llm_test.py b/assets/batch_score/components/driver/tests/e2e/mir_endpoint_llm_test.py
@@ -14,12 +14,7 @@
 
 source_dir = os.getcwd()
 gated_llm_pipeline_filepath = os.path.join(
-    source_dir,
-    "driver",
-    "tests",
-    "e2e",
-    "prs_pipeline_templates",
-    "base_llm.yml")
+    pytest.source_dir, "tests", "e2e", "prs_pipeline_templates", "base_llm.yml")
 
 RUN_NAME = "batch_score_mir_endpoint_llm_test"
 JOB_NAME = "gated_batch_score_llm"  # Should be equivalent to base_llm.yml's job name

diff --git a/assets/batch_score/components/driver/tests/e2e/serverless_endpoint_test.py b/assets/batch_score/components/driver/tests/e2e/serverless_endpoint_test.py
@@ -13,12 +13,7 @@
 # Common configuration
 source_dir = os.getcwd()
 gated_llm_pipeline_filepath = os.path.join(
-    source_dir,
-    "driver",
-    "tests",
-    "e2e",
-    "prs_pipeline_templates",
-    "base_llm.yml")
+    pytest.source_dir, "tests", "e2e", "prs_pipeline_templates", "base_llm.yml")
 
 RUN_NAME = "batch_score_aoai_endpoint_test"
 JOB_NAME = "gated_batch_score_llm"  # Should be equivalent to base_llm.yml's job name