Update torch_geometric/datasets/linkx_dataset.py

Co-authored-by: Matthias Fey <[email protected]>
pyg-team · May 2, 2022 · a0c12cc · a0c12cc
1 parent e4e5a46
commit a0c12cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/datasets/linkx_dataset.py b/torch_geometric/datasets/linkx_dataset.py
@@ -108,7 +108,7 @@ def process(self):
             x = torch.cat(xs, dim=-1)
         else:
             if 'node_feat' in mat:
-                x = torch.tensor(mat['node_feat'])
+                x = torch.from_numpy(mat['node_feat']).to(torch.float)
             elif 'features' in mat:
                 x = torch.tensor(mat['features'].todense())
             else: