From 98330df65e134f80b4202ed7679d4a5055bdcf2e Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Mon, 28 Oct 2024 11:25:13 +0100
Subject: [PATCH] Monkey patching as a desperate measure. (#2704)

* Monkey patching as a desperate measure.

* New snapshot ?
---
 integration-tests/conftest.py                 | 12 ++++++++++
 ...t_flash_starcoder_gptq_default_params.json | 22 +++++++++----------
 2 files changed, 23 insertions(+), 11 deletions(-)

diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py
index 7c082cae..c9c47766 100644
--- a/integration-tests/conftest.py
+++ b/integration-tests/conftest.py
@@ -1,3 +1,15 @@
+# ruff: noqa: E402
+import requests
+
+
+class SessionTimeoutFix(requests.Session):
+    def request(self, *args, **kwargs):
+        timeout = kwargs.pop("timeout", 120)
+        return super().request(*args, **kwargs, timeout=timeout)
+
+
+requests.sessions.Session = SessionTimeoutFix
+
 import asyncio
 import contextlib
 import json
diff --git a/integration-tests/models/__snapshots__/test_flash_starcoder_gptq/test_flash_starcoder_gptq_default_params.json b/integration-tests/models/__snapshots__/test_flash_starcoder_gptq/test_flash_starcoder_gptq_default_params.json
index ff4350b8..69938b65 100644
--- a/integration-tests/models/__snapshots__/test_flash_starcoder_gptq/test_flash_starcoder_gptq_default_params.json
+++ b/integration-tests/models/__snapshots__/test_flash_starcoder_gptq/test_flash_starcoder_gptq_default_params.json
@@ -16,52 +16,52 @@
       },
       {
         "id": 21017,
-        "logprob": -9.09375,
+        "logprob": -9.0859375,
         "text": "ometric"
       },
       {
         "id": 81,
-        "logprob": -0.2590332,
+        "logprob": -0.25927734,
         "text": "_"
       },
       {
         "id": 6009,
-        "logprob": -2.203125,
+        "logprob": -2.2109375,
         "text": "mean"
       },
       {
         "id": 26,
-        "logprob": -0.30029297,
+        "logprob": -0.2993164,
         "text": "("
       },
       {
         "id": 62,
-        "logprob": -5.6757812,
+        "logprob": -5.671875,
         "text": "L"
       },
       {
         "id": 44,
-        "logprob": -3.0898438,
+        "logprob": -3.0742188,
         "text": ":"
       },
       {
         "id": 1682,
-        "logprob": -0.67333984,
+        "logprob": -0.6777344,
         "text": " List"
       },
       {
         "id": 77,
-        "logprob": -0.3869629,
+        "logprob": -0.38354492,
         "text": "["
       },
       {
         "id": 1808,
-        "logprob": -0.92041016,
+        "logprob": -0.91845703,
         "text": "float"
       },
       {
         "id": 10794,
-        "logprob": -2.5390625,
+        "logprob": -2.5371094,
         "text": "]):"
       }
     ],
@@ -69,7 +69,7 @@
     "tokens": [
       {
         "id": 284,
-        "logprob": 0.0,
+        "logprob": -0.048583984,
         "special": false,
         "text": "\n   "
       },