diff --git a/tests/conftest.py b/tests/conftest.py index f1b290dc..5b135ec6 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -70,7 +70,9 @@ def _generate_data(): def pytest_addoption(parser): parser.addoption("--runtf", action="store_true", default=False, help="run tf tests") - parser.addoption("--runtorch", action="store_true", default=False, help="run torch tests") + parser.addoption( + "--runtorch", action="store_true", default=False, help="run torch tests" + ) def pytest_configure(config):