We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cbba176 commit 714ea28Copy full SHA for 714ea28
torch_geometric/datasets/web_qsp_dataset.py
@@ -15,12 +15,10 @@
15
get_features_for_triplets_groups,
16
)
17
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()
+from torch_geometric.llm.utils.backend_utils import (
+ retrieval_via_pcst,
+ preprocess_triplet
+)
24
25
26
class KGQABaseDataset(InMemoryDataset):
0 commit comments