Skip to content

Commit 419e0ad

Browse files
committed
fix test_cmdline
1 parent 12e89a8 commit 419e0ad

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

xinference/deploy/test/test_cmdline.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
import pytest
1919
from click.testing import CliRunner
2020

21-
from ...client import Client
21+
from ...client import RESTfulClient
2222
from ..cmdline import (
2323
list_model_registrations,
2424
model_chat,
@@ -59,7 +59,7 @@ def test_cmdline(setup, stream):
5959
"""
6060
# if use `model_launch` command to launch model, CI will fail.
6161
# So use client to launch model in temporary
62-
client = Client(endpoint)
62+
client = RESTfulClient(endpoint)
6363
model_uid = client.launch_model(
6464
model_name="orca", model_size_in_billions=3, quantization="q4_0"
6565
)

0 commit comments

Comments
 (0)