File tree 1 file changed +5
-6
lines changed
1 file changed +5
-6
lines changed Original file line number Diff line number Diff line change 56
56
57
57
np_dtype_string = np .dtype (object )
58
58
59
+ # 60 sec is the default value
60
+ NETWORK_TIMEOUT = 300.0 if TEST_VALGRIND else 60.0
61
+
59
62
60
63
class InferTest (tu .TestResultCollector ):
61
64
def _full_exact (
@@ -66,8 +69,7 @@ def _full_exact(
66
69
output0_raw ,
67
70
output1_raw ,
68
71
swap ,
69
- # 60 sec is the default value
70
- network_timeout = 60.0 ,
72
+ network_timeout = NETWORK_TIMEOUT ,
71
73
):
72
74
def _infer_exact_helper (
73
75
tester ,
@@ -88,7 +90,7 @@ def _infer_exact_helper(
88
90
skip_request_id_check = True ,
89
91
use_streaming = True ,
90
92
correlation_id = 0 ,
91
- network_timeout = 60.0 ,
93
+ network_timeout = NETWORK_TIMEOUT ,
92
94
):
93
95
for bs in (1 , batch_size ):
94
96
# model that does not support batching
@@ -550,9 +552,6 @@ def test_class_bbb(self):
550
552
output0_raw = False ,
551
553
output1_raw = False ,
552
554
swap = True ,
553
- # Increase network_timeout for TensorFlow models for
554
- # valgrind test.
555
- network_timeout = 100.0 if TEST_VALGRIND else 60.0 ,
556
555
)
557
556
558
557
def test_class_sss (self ):
You can’t perform that action at this time.
0 commit comments