From 7b4882078a74f469f2b2d79ae7650a760aee58f8 Mon Sep 17 00:00:00 2001
From: Yingge He <157551214+yinggeh@users.noreply.github.com>
Date: Mon, 30 Jun 2025 05:56:53 -0700
Subject: [PATCH] test: Remove shared memory key from the error response
 (#8269)

---
 qa/python_models/bls_memory/model.py       | 4 ++--
 qa/python_models/bls_memory_async/model.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/qa/python_models/bls_memory/model.py b/qa/python_models/bls_memory/model.py
index 69da4f440f..a55da15774 100644
--- a/qa/python_models/bls_memory/model.py
+++ b/qa/python_models/bls_memory/model.py
@@ -1,4 +1,4 @@
-# Copyright 2021-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# Copyright 2021-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions
@@ -60,7 +60,7 @@ def test_bls_out_of_memory(self):
         input0_np, infer_response = self._send_identity_tensor(
             tensor_size, self._is_decoupled
         )
-        out_of_memory_message = "Failed to increase the shared memory pool size for key"
+        out_of_memory_message = "Failed to increase the shared memory pool size"
 
         if infer_response.has_error():
             self.assertIn(out_of_memory_message, infer_response.error().message())
diff --git a/qa/python_models/bls_memory_async/model.py b/qa/python_models/bls_memory_async/model.py
index d9e676b42e..c96073eac0 100644
--- a/qa/python_models/bls_memory_async/model.py
+++ b/qa/python_models/bls_memory_async/model.py
@@ -1,4 +1,4 @@
-# Copyright 2021-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# Copyright 2021-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions
@@ -53,7 +53,7 @@ async def test_bls_out_of_memory():
     tensor_size = 256 * 1024 * 1024
     input0_np, infer_response = await _send_identity_tensor(tensor_size, is_decoupled)
 
-    out_of_memory_message = "Failed to increase the shared memory pool size for key"
+    out_of_memory_message = "Failed to increase the shared memory pool size"
 
     if infer_response.has_error():
         if not (out_of_memory_message in infer_response.error().message()):