Skip to content

Commit

Permalink
black formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
fostiropoulos committed Nov 29, 2023
1 parent 5a31a00 commit d4ae5e4
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion tests/utils/test_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}"
Expand Down

0 comments on commit d4ae5e4

Please sign in to comment.