From 414cc0b9d37a101c1144d8b1d44fed01b969eaf7 Mon Sep 17 00:00:00 2001 From: lessw2020 Date: Fri, 6 Oct 2023 05:12:42 +0000 Subject: [PATCH] yet another typedef fix attempt --- tests/test_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_utils.py b/tests/test_utils.py index abc790d9..fbf11522 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -24,7 +24,8 @@ def gpu_test(gpu_count: int = 1): """ message = f"Not enough GPUs to run the test: requires {gpu_count}" local_gpu_count: int = torch.cuda.device_count() - return pytest.mark.skipif(local_gpu_count < gpu_count, reason=message) + insufficient_gpus = local_gpu_count < gpu_count + return pytest.mark.skipif(insufficient_gpus, reason=message) def init_distributed_on_file(world_size: int, gpu_id: int, sync_file: str):