Skip to content

Commit bbbcad7

Browse files
authored
Increase network timeout for valgrind (#6324)
1 parent 40c2c45 commit bbbcad7

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

qa/L0_infer/infer_test.py

+5-6
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,9 @@
5656

5757
np_dtype_string = np.dtype(object)
5858

59+
# 60 sec is the default value
60+
NETWORK_TIMEOUT = 300.0 if TEST_VALGRIND else 60.0
61+
5962

6063
class InferTest(tu.TestResultCollector):
6164
def _full_exact(
@@ -66,8 +69,7 @@ def _full_exact(
6669
output0_raw,
6770
output1_raw,
6871
swap,
69-
# 60 sec is the default value
70-
network_timeout=60.0,
72+
network_timeout=NETWORK_TIMEOUT,
7173
):
7274
def _infer_exact_helper(
7375
tester,
@@ -88,7 +90,7 @@ def _infer_exact_helper(
8890
skip_request_id_check=True,
8991
use_streaming=True,
9092
correlation_id=0,
91-
network_timeout=60.0,
93+
network_timeout=NETWORK_TIMEOUT,
9294
):
9395
for bs in (1, batch_size):
9496
# model that does not support batching
@@ -550,9 +552,6 @@ def test_class_bbb(self):
550552
output0_raw=False,
551553
output1_raw=False,
552554
swap=True,
553-
# Increase network_timeout for TensorFlow models for
554-
# valgrind test.
555-
network_timeout=100.0 if TEST_VALGRIND else 60.0,
556555
)
557556

558557
def test_class_sss(self):

0 commit comments

Comments
 (0)