Skip to content

Commit cbba176

Browse files
authored
minor fix to triplet preproc
1 parent b0b6d4e commit cbba176

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

test/llm/test_large_graph_indexer.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
TripletLike,
1515
get_features_for_triplets,
1616
)
17+
from torch_geometric.llm.utils.backend_utils import preprocess_triplet
1718
from torch_geometric.typing import WITH_PT20
1819

1920
# create possible nodes and edges for graph
@@ -35,11 +36,6 @@ def sample_triplets(amount: int = 1) -> List[TripletLike]:
3536
return trips
3637

3738

38-
def preprocess_triplet(triplet: TripletLike) -> TripletLike:
39-
h, r, t = triplet
40-
return h.lower(), r, t.lower()
41-
42-
4339
def test_basic_collate():
4440
graphs = [sample_triplets(1000) for i in range(2)]
4541

0 commit comments

Comments
 (0)