From 7ca40d635c4757f53ec2a1260e4a9e6505f8ad49 Mon Sep 17 00:00:00 2001 From: rusty1s Date: Thu, 26 Sep 2024 09:05:57 +0200 Subject: [PATCH] Fix empty dataframe handling in WebQSPDataset --- pyproject.toml | 2 +- torch_geometric/__init__.py | 2 +- torch_geometric/datasets/web_qsp_dataset.py | 8 +++++--- 3 files changed, 7 insertions(+), 5 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index a404c7c91c58..766ad221432e 100644 --- a/pyproject.toml +++ b/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="matthias@pyg.org"}, ] diff --git a/torch_geometric/__init__.py b/torch_geometric/__init__.py index 30bc41e5c071..663993c7624d 100644 --- a/torch_geometric/__init__.py +++ b/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', diff --git a/torch_geometric/datasets/web_qsp_dataset.py b/torch_geometric/datasets/web_qsp_dataset.py index 33e12103f39b..bcc85e070920 100644 --- a/torch_geometric/datasets/web_qsp_dataset.py +++ b/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()