diff --git a/pyproject.toml b/pyproject.toml index ae9d4507a2..f36080e1ce 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -37,3 +37,7 @@ doctest_optionflags = [ [tool.pytest.ini_options] addopts = "--cov=src/peft --cov-report=term-missing" +markers = [ + "single_gpu_tests: tests that run on a single GPU", + "multi_gpu_tests: tests that run on multiple GPUs", +]