Skip to content

Commit a17b01b

Browse files
committed
don't require cuda in tests
1 parent 656a92a commit a17b01b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_radius.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
def test_radius():
77
N = 500
88
M = 100
9-
device = torch.device("cuda")
9+
device = None
1010
x = torch.randn(N, 3, device=device)
1111
y = torch.randn(M, 3, device=device)
1212
batch_x = torch.randint(2, (N,), device=device).sort().values
@@ -43,7 +43,7 @@ def test_radius():
4343
def test_radius_chunked():
4444
N = 500
4545
M = 100
46-
device = torch.device("cuda")
46+
device = None
4747
x = torch.randn(N, 3, device=device)
4848
y = torch.randn(M, 3, device=device)
4949
batch_x = torch.randint(2, (N,), device=device).sort().values

0 commit comments

Comments
 (0)