"lib/bindings/vscode:/vscode.git/clone" did not exist on "f04359cf69aef3d4ecbccd460346d6fb7081a834"
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 ...@@ -18,6 +18,7 @@ import logging
import os import os
import subprocess import subprocess
import time import time
from pathlib import Path
import pytest import pytest
import pytest_asyncio import pytest_asyncio
...@@ -28,9 +29,7 @@ logger = logging.getLogger(__name__) ...@@ -28,9 +29,7 @@ logger = logging.getLogger(__name__)
NATS_PORT = 4223 NATS_PORT = 4223
TEST_API_SERVER_MODEL_REPO_PATH = ( TEST_API_SERVER_MODEL_REPO_PATH = "integration/api_server/models"
"/workspace/worker/tests/python/integration/api_server/models"
)
def pytest_addoption(parser): def pytest_addoption(parser):
...@@ -64,7 +63,11 @@ def nats_server(log_dir): ...@@ -64,7 +63,11 @@ def nats_server(log_dir):
@pytest.fixture(scope="session") @pytest.fixture(scope="session")
def api_server(log_dir): 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" api_server_log_dir = log_dir / "api_server"
os.makedirs(api_server_log_dir, exist_ok=True) os.makedirs(api_server_log_dir, exist_ok=True)
with open(api_server_log_dir / "api_server.stdout.log", "wt") as output_: with open(api_server_log_dir / "api_server.stdout.log", "wt") as output_:
......
...@@ -284,7 +284,6 @@ def run_kserve(num_requests): ...@@ -284,7 +284,6 @@ def run_kserve(num_requests):
"(not os.path.exists('/usr/local/bin/nats-server'))", "(not os.path.exists('/usr/local/bin/nats-server'))",
reason="NATS.io not present", reason="NATS.io not present",
) )
@pytest.mark.xfail
def test_mock_disaggregated_serving_kserve(request, nats_server, workers, api_server): def test_mock_disaggregated_serving_kserve(request, nats_server, workers, api_server):
# Using a separate process to use data plane across multiple tests. # Using a separate process to use data plane across multiple tests.
p = Process(target=run_kserve, args=(1,)) 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