Commit a1f70312 authored by rusty1s's avatar rusty1s
Browse files

removed pragma

parent 38c8b3ac
......@@ -8,7 +8,7 @@ dtypes.remove(torch.short) # PyTorch scatter does not work on short types.
grad_dtypes = [torch.float, torch.double]
devices = [torch.device('cpu')]
if torch.cuda.is_available(): # pragma: no cover
if torch.cuda.is_available():
devices += [torch.device('cuda:{}'.format(torch.cuda.current_device()))]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment