Skip to content

Commit 900ab1c

Browse files
committed
tweak comment in conftest
1 parent 68c1f6f commit 900ab1c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/conftest.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,8 @@ def _setup_torch(library: Backend) -> None:
184184
if not torch.cuda.is_available():
185185
pytest.skip("no CUDA device available")
186186
torch.set_default_device("cuda")
187-
else: # TORCH
187+
else:
188+
assert library == Backend.TORCH
188189
torch.set_default_device("cpu")
189190

190191

0 commit comments

Comments
 (0)