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 b0b6d4e commit cbba176Copy full SHA for cbba176
test/llm/test_large_graph_indexer.py
@@ -14,6 +14,7 @@
14
TripletLike,
15
get_features_for_triplets,
16
)
17
+from torch_geometric.llm.utils.backend_utils import preprocess_triplet
18
from torch_geometric.typing import WITH_PT20
19
20
# create possible nodes and edges for graph
@@ -35,11 +36,6 @@ def sample_triplets(amount: int = 1) -> List[TripletLike]:
35
36
return trips
37
38
-def preprocess_triplet(triplet: TripletLike) -> TripletLike:
39
- h, r, t = triplet
40
- return h.lower(), r, t.lower()
41
-
42
43
def test_basic_collate():
44
graphs = [sample_triplets(1000) for i in range(2)]
45
0 commit comments