Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-24 23:28:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

avoid OOMs in SAM tests

Browse source
This commit is contained in:
Pierre Chapuis 2024-06-24 16:27:30 +02:00
parent 91c0fd24d8
commit 9be3580ab3
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/foundationals/segment_anything/conftest.py
View file

@ -1,9 +1,17 @@
import gc
from pathlib import Path from pathlib import Path
from warnings import warn from warnings import warn
from pytest import fixture, skip from pytest import fixture, skip
@fixture(autouse=True)
def ensure_gc():
# Avoid GPU OOMs
# See https://github.com/pytest-dev/pytest/discussions/8153#discussioncomment-214812
gc.collect()
@fixture(scope="package") @fixture(scope="package")
def ref_path(test_sam_path: Path) -> Path: def ref_path(test_sam_path: Path) -> Path:
return test_sam_path / "test_sam_ref" return test_sam_path / "test_sam_ref"