Skip to content

Commit 1713540

Browse files
committed
Fix flake8 errors
1 parent 437500e commit 1713540

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

torch_cluster/triton/radius.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,12 @@ def radius(
7878
if max_candidates == 0 or max_num_neighbors <= 0:
7979
return torch.empty(2, 0, device=x.device, dtype=torch.int64)
8080

81-
grid_out = torch.full((2, M * max_num_neighbors), -1, device=x.device, dtype=torch.int64)
81+
grid_out = torch.full(
82+
(2, M * max_num_neighbors),
83+
-1,
84+
device=x.device,
85+
dtype=torch.int64,
86+
)
8287
grid = (M,)
8388
_radius_segmented_kernel[grid](
8489
x,

0 commit comments

Comments
 (0)