Skip to content

Commit

Permalink
Replace deprecated tritongrpcclient package (#7061)
Browse files Browse the repository at this point in the history
  • Loading branch information
Tabrizian authored Apr 24, 2024
1 parent 7d1b015 commit ddd6c4b
Show file tree
Hide file tree
Showing 14 changed files with 38 additions and 38 deletions.
6 changes: 3 additions & 3 deletions qa/L0_backend_identity/identity_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,9 +32,9 @@

import numpy as np
import requests as httpreq
import tritongrpcclient as grpcclient
import tritonhttpclient as httpclient
from tritonclientutils import np_to_triton_dtype
import tritonclient.grpc as grpcclient
import tritonclient.http as httpclient
from tritonclient.utils import np_to_triton_dtype

FLAGS = None

Expand Down
6 changes: 3 additions & 3 deletions qa/L0_client_nobatch/client_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,9 +34,9 @@

import numpy as np
import test_util as tu
import tritongrpcclient
import tritonhttpclient
from tritonclientutils import InferenceServerException
import tritonclient.grpc as tritongrpcclient
import tritonclient.http as tritonhttpclient
from tritonclient.utils import InferenceServerException


class ClientNoBatchTest(tu.TestResultCollector):
Expand Down
6 changes: 3 additions & 3 deletions qa/L0_cuda_shared_memory/cuda_shared_memory_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,10 +35,10 @@

import numpy as np
import test_util as tu
import tritongrpcclient as grpcclient
import tritonhttpclient as httpclient
import tritonclient.grpc as grpcclient
import tritonclient.http as httpclient
import tritonshmutils.cuda_shared_memory as cshm
from tritonclientutils import *
from tritonclient.utils import *


class CudaSharedMemoryTest(tu.TestResultCollector):
Expand Down
6 changes: 3 additions & 3 deletions qa/L0_custom_ops/cuda_op_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@
from builtins import range

import numpy as np
import tritongrpcclient as grpcclient
import tritonhttpclient as httpclient
from tritonclientutils import np_to_triton_dtype
import tritonclient.grpc as grpcclient
import tritonclient.http as httpclient
from tritonclient.utils import np_to_triton_dtype

FLAGS = None

Expand Down
6 changes: 3 additions & 3 deletions qa/L0_custom_ops/mod_op_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@
from builtins import range

import numpy as np
import tritongrpcclient as grpcclient
import tritonhttpclient as httpclient
from tritonclientutils import np_to_triton_dtype
import tritonclient.grpc as grpcclient
import tritonclient.http as httpclient
from tritonclient.utils import np_to_triton_dtype

FLAGS = None

Expand Down
6 changes: 3 additions & 3 deletions qa/L0_custom_ops/onnx_op_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@
from builtins import range

import numpy as np
import tritongrpcclient as grpcclient
import tritonhttpclient as httpclient
from tritonclientutils import np_to_triton_dtype
import tritonclient.grpc as grpcclient
import tritonclient.http as httpclient
from tritonclient.utils import np_to_triton_dtype

FLAGS = None

Expand Down
6 changes: 3 additions & 3 deletions qa/L0_custom_ops/zero_out_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@
from builtins import range

import numpy as np
import tritongrpcclient as grpcclient
import tritonhttpclient as httpclient
from tritonclientutils import np_to_triton_dtype
import tritonclient.grpc as grpcclient
import tritonclient.http as httpclient
from tritonclient.utils import np_to_triton_dtype

FLAGS = None

Expand Down
6 changes: 3 additions & 3 deletions qa/L0_large_payload/large_payload_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,9 +35,9 @@

import numpy as np
import test_util as tu
import tritongrpcclient as grpcclient
import tritonhttpclient as httpclient
from tritonclientutils import InferenceServerException, np_to_triton_dtype
import tritonclient.grpc as grpcclient
import tritonclient.http as httpclient
from tritonclient.utils import InferenceServerException, np_to_triton_dtype


class LargePayLoadTest(tu.TestResultCollector):
Expand Down
6 changes: 3 additions & 3 deletions qa/L0_nullchar_string/nullchar_string_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,9 +28,9 @@
import argparse

import numpy as np
import tritongrpcclient as grpcclient
import tritonhttpclient as httpclient
from tritonclientutils import np_to_triton_dtype
import tritonclient.grpc as grpcclient
import tritonclient.http as httpclient
from tritonclient.utils import np_to_triton_dtype

FLAGS = None

Expand Down
4 changes: 2 additions & 2 deletions qa/L0_rate_limiter/rate_limiter_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,8 +38,8 @@

import numpy as np
import sequence_util as su
import tritongrpcclient as grpcclient
from tritonclientutils import *
import tritonclient.grpc as grpcclient
from tritonclient.utils import *

_inference_count = 80
_inference_concurrency = 8
Expand Down
6 changes: 3 additions & 3 deletions qa/L0_repoagent_checksum/identity_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,9 +30,9 @@
import sys

import numpy as np
import tritongrpcclient as grpcclient
import tritonhttpclient as httpclient
from tritonclientutils import np_to_triton_dtype
import tritonclient.grpc as grpcclient
import tritonclient.http as httpclient
from tritonclient.utils import np_to_triton_dtype

FLAGS = None

Expand Down
4 changes: 2 additions & 2 deletions qa/L0_sequence_stress/sequence_stress.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,8 +39,8 @@

import numpy as np
import test_util as tu
import tritongrpcclient as grpcclient
from tritonclientutils import np_to_triton_dtype
import tritonclient.grpc as grpcclient
from tritonclient.utils import np_to_triton_dtype

if sys.version_info >= (3, 0):
import queue
Expand Down
6 changes: 3 additions & 3 deletions qa/L0_server_status/server_status_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,9 +36,9 @@
import infer_util as iu
import numpy as np
import test_util as tu
import tritongrpcclient as grpcclient
import tritonhttpclient as httpclient
from tritonclientutils import *
import tritonclient.grpc as grpcclient
import tritonclient.http as httpclient
from tritonclient.utils import *


class ServerMetadataTest(tu.TestResultCollector):
Expand Down
2 changes: 1 addition & 1 deletion qa/L0_trt_shape_tensors/trt_shape_tensor_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@
import numpy as np
import sequence_util as su
import test_util as tu
import tritongrpcclient as grpcclient
import tritonclient.grpc as grpcclient

TEST_SYSTEM_SHARED_MEMORY = bool(int(os.environ.get("TEST_SYSTEM_SHARED_MEMORY", 0)))

Expand Down

0 comments on commit ddd6c4b

Please sign in to comment.