Correctly reshape node features in BrcaTcga dataset (pyg-team#7971)

kgajdamo · Sep 4, 2023 · cf60b5f · cf60b5f
1 parent 6db1453
commit cf60b5f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/datasets/brca_tgca.py b/torch_geometric/datasets/brca_tgca.py
@@ -92,7 +92,7 @@ def process(self):
 
         data_list = []
         for x, y in zip(graph_feat, graph_label):
-            data = Data(x=x, edge_index=edge_index, y=y)
+            data = Data(x=x.view(-1, 1), edge_index=edge_index, y=y)
 
             if self.pre_filter is not None and not self.pre_filter(data):
                 continue