Fix BitcoinOTC processing (#6470)

pyg-team · Jan 19, 2023 · 6fcf347 · 6fcf347
1 parent 5c3ad1e
commit 6fcf347
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/datasets/bitcoin_otc.py b/torch_geometric/datasets/bitcoin_otc.py
@@ -70,7 +70,7 @@ def process(self):
             edge_index = edge_index.t().contiguous()
             num_nodes = edge_index.max().item() + 1
 
-            edge_attr = [float(line[2]) for line in data]
+            edge_attr = [int(line[2]) for line in data]
             edge_attr = torch.tensor(edge_attr, dtype=torch.long)
 
             stamps = [int(float(line[3])) for line in data]