Fix L0_implicit_state (#5595)

triton-inference-server · Apr 5, 2023 · c47b239 · c47b239
1 parent cac40ca
commit c47b239
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/qa/L0_implicit_state/implicit_state.py b/qa/L0_implicit_state/implicit_state.py
@@ -133,9 +133,10 @@ def test_request_output_not_allowed(self):
                     sequence_id=1,
                     sequence_start=True,
                     sequence_end=True)
-            self.assertTrue(
-                str(e.exception).startswith(
-                    "unexpected inference output 'OUTPUT_STATE' for model"))
+            self.assertIn(
+                "unexpected inference output 'OUTPUT_STATE' for model",
+                str(e.exception)
+            )
 
     def test_request_output(self):
         triton_client = tritonhttpclient.InferenceServerClient("localhost:8000")