Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove @slow decorator in favor of @pytest.mark.slow.

Browse Source
This commit is contained in:
Ryan Dick
2023-10-06 18:26:06 -04:00
parent 7870b90717
commit e0e001758a
3 changed files with 6 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
invokeai/backend/util/test_utils.py
View File

@ -11,16 +11,6 @@ from invokeai.backend.model_management.model_manager import ModelInfo
from invokeai.backend.model_management.models.base import BaseModelType, ModelNotFoundException, ModelType, SubModelType
def slow(test_case):
"""Decorator for slow tests.
Tests should be marked as slow if they download a model, run model inference, or do anything else slow. To judge
whether a test is 'slow', consider how it would perform in a CPU-only environment with a low-bandwidth internet
connection.
"""
return pytest.mark.slow(test_case)
@pytest.fixture(scope="session")
def torch_device():
return "cuda" if torch.cuda.is_available() else "cpu"