huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-20 06:12:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Simplify logs2. (#3045)

Browse Source 
* Simplify logs2.

* Changing the scope from module to session to fix the event_loop issue.
This commit is contained in:
Nicolas Patry 2025-02-21 10:03:40 +01:00 committed by GitHub
parent 06dfe9abfe
commit 142a49a80d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
integration-tests/conftest.py
View File

@ -81,6 +81,8 @@ def container_log(request: SubRequest):
if request.session.testsfailed: if request.session.testsfailed:
error_log.seek(0) error_log.seek(0)
print(error_log.read(), file=sys.stderr) print(error_log.read(), file=sys.stderr)
else:
error_log.truncate(0)
class ResponseComparator(JSONSnapshotExtension): class ResponseComparator(JSONSnapshotExtension):
@ -350,13 +352,13 @@ def ignore_logprob_response_snapshot(snapshot):
return snapshot.use_extension(IgnoreLogProbResponseComparator) return snapshot.use_extension(IgnoreLogProbResponseComparator)
@pytest.fixture(scope="module") @pytest.fixture(scope="session")
def error_log(): def error_log():
with tempfile.TemporaryFile("w+") as tmp: with tempfile.TemporaryFile("w+") as tmp:
yield tmp yield tmp
@pytest.fixture(scope="module") @pytest.fixture(scope="session")
async def launcher(error_log): async def launcher(error_log):
@contextlib.contextmanager @contextlib.contextmanager
def local_launcher( def local_launcher(