From 7ad5d9a222df5146b254c54182a956d8933457b4 Mon Sep 17 00:00:00 2001 From: Richard Brown <33289025+rijobro@users.noreply.github.com> Date: Fri, 22 Jan 2021 13:43:10 +0000 Subject: [PATCH] remove decorator double negative Signed-off-by: Richard Brown <33289025+rijobro@users.noreply.github.com> --- tests/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/utils.py b/tests/utils.py index 20de17bbff..d73cb5fdc7 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -85,14 +85,14 @@ def skip_if_no_cpp_extention(obj): """ Skip the unit tests if the cpp extention isnt available """ - return unittest.skipIf(not USE_COMPILED, "Skipping cpp extention tests")(obj) + return unittest.skipUnless(USE_COMPILED, "Skipping cpp extention tests")(obj) def skip_if_no_cuda(obj): """ Skip the unit tests if torch.cuda.is_available is False """ - return unittest.skipIf(not torch.cuda.is_available(), "Skipping CUDA-based tests")(obj) + return unittest.skipUnless(torch.cuda.is_available(), "Skipping CUDA-based tests")(obj) def skip_if_windows(obj):