huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 20:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Monkey patching as a desperate measure.

Browse Source
This commit is contained in:
Nicolas Patry 2024-10-28 11:10:00 +01:00
parent 513d19b955
commit 7daf27ca58
No known key found for this signature in database
GPG Key ID: D2920555C90F704C
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
integration-tests/conftest.py
View File

@ -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 asyncio
import contextlib import contextlib
import json import json