ensure data is torch in torch_geometric.data

rendeirolab · Aug 28, 2024 · 22402f1 · 22402f1
1 parent 33facab
commit 22402f1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/wsi/wsi.py b/wsi/wsi.py
@@ -1907,5 +1907,9 @@ def as_torch_geometric_data(
             assert coords is not None, "coords must be provided when feats is not None"
 
         edge_index = self.get_tile_graph()
-        data = Data(x=feats, edge_index=edge_index, pos=coords)
+        data = Data(
+            x=torch.tensor(feats),
+            edge_index=torch.tensor(edge_index),
+            pos=torch.tensor(coords),
+        )
         return data