Removing warnings appearing in several distributed tests.

dmlc · Jul 31, 2024 · 5dd0c75 · 5dd0c75
1 parent 65f85b5
commit 5dd0c75
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/distributed/test_distributed_sampling.py b/tests/distributed/test_distributed_sampling.py
@@ -512,7 +512,7 @@ def start_hetero_sample_client(
     assert "feat" not in dist_graph.nodes["n2"].data
     assert "feat" not in dist_graph.nodes["n3"].data
     nodes = {
-        k: torch.tensor(v, dtype=dist_graph.idtype) for k, v in nodes.items()
+        k: v.type(dist_graph.idtype).clone().detach() for k, v in nodes.items()
     }
     if gpb is None:
         gpb = dist_graph.get_partition_book()
@@ -553,7 +553,7 @@ def start_hetero_etype_sample_client(
     assert "feat" not in dist_graph.nodes["n2"].data
     assert "feat" not in dist_graph.nodes["n3"].data
     nodes = {
-        k: torch.tensor(v, dtype=dist_graph.idtype) for k, v in nodes.items()
+        k: v.type(dist_graph.idtype).clone().detach() for k, v in nodes.items()
     }
 
     if (not use_graphbolt) and dist_graph.local_partition is not None:
@@ -915,7 +915,7 @@ def start_bipartite_sample_client(
     assert "feat" in dist_graph.nodes["user"].data
     assert "feat" in dist_graph.nodes["game"].data
     nodes = {
-        k: torch.tensor(v, dtype=dist_graph.idtype) for k, v in nodes.items()
+        k: v.type(dist_graph.idtype).clone().detach() for k, v in nodes.items()
     }
     if gpb is None:
         gpb = dist_graph.get_partition_book()
@@ -951,7 +951,7 @@ def start_bipartite_etype_sample_client(
     assert "feat" in dist_graph.nodes["user"].data
     assert "feat" in dist_graph.nodes["game"].data
     nodes = {
-        k: torch.tensor(v, dtype=dist_graph.idtype) for k, v in nodes.items()
+        k: v.type(dist_graph.idtype).clone().detach() for k, v in nodes.items()
     }
 
     if not use_graphbolt and dist_graph.local_partition is not None: