fix test

dmlc · Feb 24, 2023 · 1cdf545 · 1cdf545
1 parent a4a2f46
commit 1cdf545
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/common/test_homophily.py b/tests/python/common/test_homophily.py
@@ -40,6 +40,6 @@ def test_linkx_homophily(idtype):
     # IfChangeThenChange: python/dgl/homophily.py
     # Update the docstring example.
     device = F.ctx()
-    graph = dgl.graph(([0, 1, 2, 3], [1, 2, 0, 4]))
+    graph = dgl.graph(([0, 1, 2, 3], [1, 2, 0, 4]), device=device)
     y = F.tensor([0, 0, 0, 0, 1])
     assert dgl.linkx_homophily(graph, y) == 0.19999998807907104