From a1953baef6c9c585604cdb86ebf9834f8aa514ef Mon Sep 17 00:00:00 2001 From: Benjamin Bossan Date: Mon, 17 Jul 2023 12:00:08 +0200 Subject: [PATCH] FIX: Removes warnings about unknown pytest marker (#715) This is a low prio PR but it solves an annoyance. Right now, when running tests, the output is spammed by messages like: > PytestUnknownMarkWarning: Unknown pytest.mark.multi_gpu_tests - is this a typo? ... This makes it more difficult to see the actually relevant information. This PR fixes this by registering the two pytest markers we use, thus removing the warnings. --- pyproject.toml | 4 ++++ 1 file changed, 4 insertions(+) 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", +]