pragma

acd094e7 · rusty1s · 7f186c26 · acd094e7
Commit acd094e7 authored Jul 26, 2019 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torch_cluster/rw.py torch_cluster/rw.py +1 -1

No files found.
--- a/torch_cluster/rw.py
+++ b/torch_cluster/rw.py
@@ -46,7 +46,7 @@ def random_walk(row, col, start, walk_length, p=1, q=1, coalesced=False,

    start = start.flatten()

-    if row.is_cuda:
+    if row.is_cuda:  # pragma: no cover
        return torch_cluster.rw_cuda.rw(row, col, start, walk_length, p, q,
                                        num_nodes)
    else: