From d4ae5e4d49490bed930b1a66dc971fa47c6246f2 Mon Sep 17 00:00:00 2001 From: Iordanis Fostiropoulos Date: Wed, 29 Nov 2023 20:01:06 +0000 Subject: [PATCH] black formatting --- tests/utils/test_base.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/utils/test_base.py b/tests/utils/test_base.py index 003e83e4..9be4a556 100644 --- a/tests/utils/test_base.py +++ b/tests/utils/test_base.py @@ -24,7 +24,6 @@ def test_set_seed(): def test_parse_device(): assert base.parse_device("cpu") == "cpu" if torch.cuda.is_available(): - gpu_number = torch.cuda.device_count() for i in range(gpu_number): assert base.parse_device(i) == f"cuda:{i}"