Adding timeout to requests in tests

Author: ivan-khvostishkov

tests/source_dir/inference_tf/inference.py

@@ -20,7 +20,7 @@ def handler(data, context):
         (bytes, string): data to return to client, (optional) response content type
     """
     processed_input = _process_input(data, context)
-    response = requests.post(context.rest_uri, data=processed_input)
+    response = requests.post(context.rest_uri, data=processed_input, timeout=5)
     return _process_output(response, context)
 
 

tests/source_dir/inference_tf_clean/inference.py

@@ -13,7 +13,7 @@ def handler(data, context):
         (bytes, string): data to return to client, (optional) response content type
     """
     processed_input = _process_input(data, context)
-    response = requests.post(context.rest_uri, data=processed_input)
+    response = requests.post(context.rest_uri, data=processed_input, timeout=5)
     return _process_output(response, context)
 
 

tests/source_dir/inference_tf_clean_model2/inference.py

@@ -13,7 +13,7 @@ def handler(data, context):
         (bytes, string): data to return to client, (optional) response content type
     """
     processed_input = _process_input(data, context)
-    response = requests.post(context.rest_uri, data=processed_input)
+    response = requests.post(context.rest_uri, data=processed_input, timeout=5)
     return _process_output(response, context)
 
 

tests/source_dir/inference_tf_model2/inference.py

@@ -20,7 +20,7 @@ def handler(data, context):
         (bytes, string): data to return to client, (optional) response content type
     """
     processed_input = _process_input(data, context)
-    response = requests.post(context.rest_uri, data=processed_input)
+    response = requests.post(context.rest_uri, data=processed_input, timeout=5)
     return _process_output(response, context)