Merge branch 'master' into cov/llm/vit

Author: xnuohz

.pre-commit-config.yaml

@@ -35,7 +35,7 @@ repos:
         args: [-d, '{extends: default, rules: {line-length: disable, document-start: disable, truthy: {level: error}, braces: {max-spaces-inside: 1}}}']
 
   - repo: https://github.com/asottile/pyupgrade
-    rev: v3.21.0
+    rev: v3.21.2
     hooks:
       - id: pyupgrade
         name: Upgrade Python syntax
@@ -75,7 +75,7 @@ repos:
         additional_dependencies: [Flake8-pyproject]
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.14.3
+    rev: v0.14.7
     hooks:
       - id: ruff
         name: Ruff formatting
@@ -92,7 +92,7 @@ repos:
           - mdformat_footnote
 
   - repo: https://github.com/sphinx-contrib/sphinx-lint
-    rev: v1.0.1
+    rev: v1.0.2
     hooks:
       - id: sphinx-lint
         name: Check Sphinx

examples/llm/nvtx_examples/README.md

@@ -208,6 +208,8 @@ filterwarnings = [
     "error::DeprecationWarning",
     # TODO(rishipuri98): Remove usage of `torch_geometric.distributed` from `torch_geometric.llm`
     "ignore:.*torch_geometric.distributed.*:DeprecationWarning",
+    # Filter `torch.jit.*` deprication warnings:
+    "ignore:.*torch.jit.*:DeprecationWarning",
 ]
 markers = [
     "rag: mark test as RAG test",

examples/llm/nvtx_examples/nvtx_rag_backend_example.py

@@ -14,12 +14,17 @@
     TripletLike,
     get_features_for_triplets,
 )
+from torch_geometric.llm.utils.backend_utils import preprocess_triplet
 from torch_geometric.typing import WITH_PT20
 
 # create possible nodes and edges for graph
 strkeys = string.ascii_letters + string.digits
-NODE_POOL = list({"".join(random.sample(strkeys, 10)) for i in range(1000)})
-EDGE_POOL = list({"".join(random.sample(strkeys, 10)) for i in range(50)})
+NODE_POOL = list(
+    {"".join(random.sample(strkeys, 10)).lower()
+     for i in range(1000)})
+EDGE_POOL = list(
+    {"".join(random.sample(strkeys, 10)).lower()
+     for i in range(50)})
 
 
 def featurize(s: str) -> int:
@@ -35,11 +40,6 @@ def sample_triplets(amount: int = 1) -> List[TripletLike]:
     return trips
 
 
-def preprocess_triplet(triplet: TripletLike) -> TripletLike:
-    h, r, t = triplet
-    return h.lower(), r, t.lower()
-
-
 def test_basic_collate():
     graphs = [sample_triplets(1000) for i in range(2)]
 

examples/llm/nvtx_examples/nvtx_run.sh

@@ -15,12 +15,10 @@
     get_features_for_triplets_groups,
 )
 from torch_geometric.llm.models import SentenceTransformer
-from torch_geometric.llm.utils.backend_utils import retrieval_via_pcst
-
-
-def preprocess_triplet(triplet: TripletLike) -> TripletLike:
-    h, r, t = triplet
-    return str(h).lower(), str(r).lower(), str(t).lower()
+from torch_geometric.llm.utils.backend_utils import (
+    preprocess_triplet,
+    retrieval_via_pcst,
+)
 
 
 class KGQABaseDataset(InMemoryDataset):