ctransformers

1
import pytest
2

3

4
def pytest_addoption(parser):
5
    parser.addoption(
6
        "--lib",
7
        action="store",
8
        choices=("avx2", "avx", "basic", "cuda", "none"),
9
    )
10

11

12
@pytest.fixture
13
def lib(request):
14
    lib = request.config.getoption("--lib")
15
    return lib if lib != "none" else None
16