tests/conftest.py

"""Pytest configuration for InstaNovo Gradio tests."""

import pytest


def pytest_configure(config):
    """Configure pytest markers."""
    config.addinivalue_line(
        "markers", "slow: marks tests as slow (deselect with '-m \"not slow\"')"
    )


@pytest.fixture(scope="session")
def device():
    """Get the device for testing."""
    import torch
    return "cuda" if torch.cuda.is_available() else "cpu"