Fixing error message substr check

triton-inference-server · Oct 1, 2024 · 0da3160 · 0da3160
1 parent 977774b
commit 0da3160
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/qa/L0_python_api/test_kserve.py b/qa/L0_python_api/test_kserve.py
@@ -242,17 +242,18 @@ def callback(user_data, result, error):
  time.sleep(1)
 
  # Depending on when gRPC frontend shut down StatusCode can vary
- acceptable_failure_msgs = set(
- [
- "[StatusCode.CANCELLED] CANCELLED",
- "[StatusCode.UNAVAILABLE] failed to connect to all addresses",
- ]
- )
+ acceptable_failure_msgs = [
+ "[StatusCode.CANCELLED] CANCELLED",
+ "[StatusCode.UNAVAILABLE] failed to connect to all addresses",
+ ]
 
  assert (
  len(user_data) == 1
  and isinstance(user_data[0], InferenceServerException)
- and str(user_data[0]) in acceptable_failure_msgs
+ and any(
+ failure_msg in str(user_data[0])
+ for failure_msg in acceptable_failure_msgs
+ )
  )
 
  teardown_client(grpc_client)