Skip to content

Commit 44e4168

Browse files
mc-nvyinggeh
andauthored
test: Remove shared memory key from the error response (#8269) (#8308)
Co-authored-by: Yingge He <[email protected]>
1 parent 89888ce commit 44e4168

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

qa/python_models/bls_memory/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2021-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+
# Copyright 2021-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
#
33
# Redistribution and use in source and binary forms, with or without
44
# modification, are permitted provided that the following conditions
@@ -60,7 +60,7 @@ def test_bls_out_of_memory(self):
6060
input0_np, infer_response = self._send_identity_tensor(
6161
tensor_size, self._is_decoupled
6262
)
63-
out_of_memory_message = "Failed to increase the shared memory pool size for key"
63+
out_of_memory_message = "Failed to increase the shared memory pool size"
6464

6565
if infer_response.has_error():
6666
self.assertIn(out_of_memory_message, infer_response.error().message())

qa/python_models/bls_memory_async/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2021-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+
# Copyright 2021-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
#
33
# Redistribution and use in source and binary forms, with or without
44
# modification, are permitted provided that the following conditions
@@ -53,7 +53,7 @@ async def test_bls_out_of_memory():
5353
tensor_size = 256 * 1024 * 1024
5454
input0_np, infer_response = await _send_identity_tensor(tensor_size, is_decoupled)
5555

56-
out_of_memory_message = "Failed to increase the shared memory pool size for key"
56+
out_of_memory_message = "Failed to increase the shared memory pool size"
5757

5858
if infer_response.has_error():
5959
if not (out_of_memory_message in infer_response.error().message()):

0 commit comments

Comments
 (0)