Commit db48e835 authored by Tanmay Verma's avatar Tanmay Verma Committed by GitHub
Browse files

fix: Fix the mock disaggregated serving test (#87)

parent b34c6f77
......@@ -18,6 +18,7 @@ import logging
import os
import subprocess
import time
from pathlib import Path
import pytest
import pytest_asyncio
......@@ -28,9 +29,7 @@ logger = logging.getLogger(__name__)
NATS_PORT = 4223
TEST_API_SERVER_MODEL_REPO_PATH = (
"/workspace/worker/tests/python/integration/api_server/models"
)
TEST_API_SERVER_MODEL_REPO_PATH = "integration/api_server/models"
def pytest_addoption(parser):
......@@ -64,7 +63,11 @@ def nats_server(log_dir):
@pytest.fixture(scope="session")
def api_server(log_dir):
command = ["tritonserver", "--model-store", str(TEST_API_SERVER_MODEL_REPO_PATH)]
command = [
"tritonserver",
"--model-store",
str(Path(__file__).parent.resolve() / TEST_API_SERVER_MODEL_REPO_PATH),
]
api_server_log_dir = log_dir / "api_server"
os.makedirs(api_server_log_dir, exist_ok=True)
with open(api_server_log_dir / "api_server.stdout.log", "wt") as output_:
......
......@@ -284,7 +284,6 @@ def run_kserve(num_requests):
"(not os.path.exists('/usr/local/bin/nats-server'))",
reason="NATS.io not present",
)
@pytest.mark.xfail
def test_mock_disaggregated_serving_kserve(request, nats_server, workers, api_server):
# Using a separate process to use data plane across multiple tests.
p = Process(target=run_kserve, args=(1,))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment