Fix empty dataframe handling in WebQSPDataset

Author: rusty1s

pyproject.toml

@@ -4,7 +4,7 @@ build-backend="flit_core.buildapi"
 
 [project]
 name="torch-geometric"
-version="2.6.0"
+version="2.6.1"
 authors=[
     {name="Matthias Fey", email="[email protected]"},
 ]

torch_geometric/__init__.py

@@ -30,7 +30,7 @@
 contrib = LazyLoader('contrib', globals(), 'torch_geometric.contrib')
 graphgym = LazyLoader('graphgym', globals(), 'torch_geometric.graphgym')
 
-__version__ = '2.6.0'
+__version__ = '2.6.1'
 
 __all__ = [
     'Index',

torch_geometric/datasets/web_qsp_dataset.py

@@ -196,8 +196,10 @@ def process(self) -> None:
                 nodes = pd.DataFrame([{
                     "node_id": v,
                     "node_attr": k,
-                } for k, v in raw_nodes.items()])
-                edges = pd.DataFrame(raw_edges)
+                } for k, v in raw_nodes.items()],
+                                     columns=["node_id", "node_attr"])
+                edges = pd.DataFrame(raw_edges,
+                                     columns=["src", "edge_attr", "dst"])
 
                 nodes.node_attr = nodes.node_attr.fillna("")
                 x = model.encode(
@@ -213,7 +215,7 @@ def process(self) -> None:
                 edge_index = torch.tensor([
                     edges.src.tolist(),
                     edges.dst.tolist(),
-                ])
+                ], dtype=torch.long)
 
                 question = f"Question: {example['question']}\nAnswer: "
                 label = ('|').join(example['answer']).lower()