From 57803e03fc5df1f3ae7698d80760fe9e701fdc88 Mon Sep 17 00:00:00 2001 From: "pixeebot[bot]" <104101892+pixeebot[bot]@users.noreply.github.com> Date: Thu, 30 May 2024 18:10:00 -0600 Subject: [PATCH] Add timeout to `requests` calls (#458) * Add timeout to `requests` calls * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * trailing comma * fix test --------- Co-authored-by: pixeebot[bot] <104101892+pixeebot[bot]@users.noreply.github.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Taylor Jackle Spriggs --- sigopt/run_context.py | 1 + test/runs/test_context.py | 1 + 2 files changed, 2 insertions(+) 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, )