Skip to content

Commit 714ea28

Browse files
authored
Update web_qsp_dataset.py
1 parent cbba176 commit 714ea28

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

torch_geometric/datasets/web_qsp_dataset.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,10 @@
1515
get_features_for_triplets_groups,
1616
)
1717
from torch_geometric.llm.models import SentenceTransformer
18-
from torch_geometric.llm.utils.backend_utils import retrieval_via_pcst
19-
20-
21-
def preprocess_triplet(triplet: TripletLike) -> TripletLike:
22-
h, r, t = triplet
23-
return str(h).lower(), str(r).lower(), str(t).lower()
18+
from torch_geometric.llm.utils.backend_utils import (
19+
retrieval_via_pcst,
20+
preprocess_triplet
21+
)
2422

2523

2624
class KGQABaseDataset(InMemoryDataset):

0 commit comments

Comments
 (0)