From 777761d2a0f5d2926cb00cddd275461a1d635b85 Mon Sep 17 00:00:00 2001
From: krishung5 <krish@nvidia.com>
Date: Wed, 13 Sep 2023 16:58:42 -0700
Subject: [PATCH] Increase network timeout for valgrind

---
 qa/L0_infer/infer_test.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/qa/L0_infer/infer_test.py b/qa/L0_infer/infer_test.py
index d97803b17d3..cc264ac00df 100755
--- a/qa/L0_infer/infer_test.py
+++ b/qa/L0_infer/infer_test.py
@@ -56,6 +56,8 @@
 
 np_dtype_string = np.dtype(object)
 
+# 60 sec is the default value
+NETWORK_TIMEOUT = 300.0 if TEST_VALGRIND else 60.0
 
 class InferTest(tu.TestResultCollector):
     def _full_exact(
@@ -66,8 +68,7 @@ def _full_exact(
         output0_raw,
         output1_raw,
         swap,
-        # 60 sec is the default value
-        network_timeout=60.0,
+        network_timeout=NETWORK_TIMEOUT,
     ):
         def _infer_exact_helper(
             tester,
@@ -88,7 +89,7 @@ def _infer_exact_helper(
             skip_request_id_check=True,
             use_streaming=True,
             correlation_id=0,
-            network_timeout=60.0,
+            network_timeout=NETWORK_TIMEOUT,
         ):
             for bs in (1, batch_size):
                 # model that does not support batching
@@ -550,9 +551,6 @@ def test_class_bbb(self):
                 output0_raw=False,
                 output1_raw=False,
                 swap=True,
-                # Increase network_timeout for TensorFlow models for
-                # valgrind test.
-                network_timeout=100.0 if TEST_VALGRIND else 60.0,
             )
 
         def test_class_sss(self):