Fix empty dataframe handling in WebQSPDataset

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()