Browse Source
Update triton_client iterface
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
client.py
|
|
@ -18,7 +18,7 @@ class Client(PyOperator): |
|
|
""" |
|
|
""" |
|
|
def __init__(self, url, model_name='pipeline'): |
|
|
def __init__(self, url, model_name='pipeline'): |
|
|
super().__init__() |
|
|
super().__init__() |
|
|
self._client = triton_client(url, model_name) |
|
|
|
|
|
|
|
|
self._client = triton_client.Client(url, model_name) |
|
|
|
|
|
|
|
|
def __call__(self, *data): |
|
|
def __call__(self, *data): |
|
|
""" |
|
|
""" |
|
|
|
