diff --git a/sigopt/run_context.py b/sigopt/run_context.py index 81053bd5..88fb848a 100644 --- a/sigopt/run_context.py +++ b/sigopt/run_context.py @@ -459,6 +459,7 @@ def _log_image(self, name, payload): upload_info["url"], headers=upload_info["headers"], data=image_data, + timeout=60, ) response.raise_for_status() diff --git a/test/runs/test_context.py b/test/runs/test_context.py index f8ab2e6d..41adabfa 100644 --- a/test/runs/test_context.py +++ b/test/runs/test_context.py @@ -243,4 +243,5 @@ def test_log_image(self, run_context): "https://test.sigopt.ninja/upload_file", headers={"X-Testing": "value"}, data=image_data, + timeout=mock.ANY, )