tqdm for triplets

pyg-team · Jan 10, 2025 · 139e0f8 · 139e0f8
1 parent 02c53ed
commit 139e0f8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_geometric/data/large_graph_indexer.py b/torch_geometric/data/large_graph_indexer.py
@@ -163,7 +163,7 @@ def from_triplets(
 
         Args:
             triplets (KnowledgeGraphLike): Series of triplets representing
-                knowledge graph relations.
+                knowledge graph relations. Example: [("cats", "eat", dogs")].
             pre_transform (Optional[Callable[[TripletLike], TripletLike]]):
                 Optional preprocessing function to apply to triplets.
                 Defaults to None.
@@ -182,7 +182,7 @@ def apply_transform(
                 for trip in trips:
                     yield pre_transform(trip)
 
-            triplets = apply_transform(triplets)
+            triplets = list(apply_transform(triplets))
 
         for h, r, t in tqdm(triplets, desc="Indexing Triples"):