From 3307dc0f4752b1193ba99e1975b6d4f271ad1b67 Mon Sep 17 00:00:00 2001
From: shiyu22 <shiyu.chen@zilliz.com>
Date: Wed, 8 Feb 2023 15:41:13 +0800
Subject: [PATCH] Update triton_client iterface

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
---
 client.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/client.py b/client.py
index 447e985..3e43fd0 100644
--- a/client.py
+++ b/client.py
@@ -18,7 +18,7 @@ class Client(PyOperator):
     """
     def __init__(self, url, model_name='pipeline'):
         super().__init__()
-        self._client = triton_client(url, model_name)
+        self._client = triton_client.Client(url, model_name)
 
     def __call__(self, *data):
         """