pyg-nightly 2.7.0.dev20250909__py3-none-any.whl → 2.7.0.dev20250910__py3-none-any.whl

{pyg_nightly-2.7.0.dev20250909.dist-info → pyg_nightly-2.7.0.dev20250910.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyg-nightly
-Version: 2.7.0.dev20250909
+Version: 2.7.0.dev20250910
 Summary: Graph Neural Network Library for PyTorch
 Keywords: deep-learning,pytorch,geometric-deep-learning,graph-neural-networks,graph-convolutional-networks
 Author-email: Matthias Fey <matthias@pyg.org>

{pyg_nightly-2.7.0.dev20250909.dist-info → pyg_nightly-2.7.0.dev20250910.dist-info}/RECORD

@@ -1,4 +1,4 @@
-torch_geometric/__init__.py,sha256=b9w_XN3wm4unZ7jxNsQ327GPF9zKTV44a5HfC4fSUvw,2292
+torch_geometric/__init__.py,sha256=6DYZWRpxtgSBbAwl2eIiEylFG48gBD1uBNmuBQhahx4,2292
 torch_geometric/_compile.py,sha256=9yqMTBKatZPr40WavJz9FjNi7pQj8YZAZOyZmmRGXgc,1351
 torch_geometric/_onnx.py,sha256=ODB_8cwFUiwBUjngXn6-K5HHb7IDul7DDXuuGX7vj_0,8178
 torch_geometric/backend.py,sha256=lVaf7aLoVaB3M-UcByUJ1G4T4FOK6LXAg0CF4W3E8jo,1575
@@ -267,10 +267,10 @@ torch_geometric/io/ply.py,sha256=NdeTtr79vJ1HS37ZV2N61EUmA5NGJd2I6cUj1Pg7Ypg,489
 torch_geometric/io/sdf.py,sha256=H2PC6dSW9Kncc1ulb0UN0JnTRT93NY2fY8lf6K4hb50,1165
 torch_geometric/io/tu.py,sha256=-v5Ago7DfmGTRBtB5RZFvmv4XpLnKKnk-NOnxlHtB_c,4881
 torch_geometric/io/txt_array.py,sha256=LDeX2qtlNKW-kVe-wpnskMwAdXQp1jVCGQnrJce7Smg,910
-torch_geometric/llm/__init__.py,sha256=ikGCkUbn_gjlq6GPSDu43sErVN8syBetNNasqtn9X74,217
+torch_geometric/llm/__init__.py,sha256=qOCyBeWO8vcUTI71DLv1-lrssWoAqIyE1JTuzHM4rDU,221
 torch_geometric/llm/large_graph_indexer.py,sha256=ip5wU7Rj4S1tFc53AqAtvyvGqeVC7AJrVMEbRTHyz54,28429
 torch_geometric/llm/rag_loader.py,sha256=lsUt5kvHGbQzxOuWubBVB0RkMcTi6b4RNRuvsArw3Ys,5641
-torch_geometric/llm/models/__init__.py,sha256=nzO63kzvbWNRicrxvmRl0m622WEAZ6_x8XI_hCocTH4,537
+torch_geometric/llm/models/__init__.py,sha256=Xb2GacStLOYCAu34NNzG-zog-zapXh1asl3S6_eLMsc,547
 torch_geometric/llm/models/g_retriever.py,sha256=SeW0rlrlzMhN3aVslhq_GUyUgS4sVw_nMAT5YiXzZd8,9072
 torch_geometric/llm/models/git_mol.py,sha256=m1YJb6Xb2i6j9wEqHzqE4YBWMr9i1CJpMp6T-E24fsA,12680
 torch_geometric/llm/models/glem.py,sha256=GlL_I63g-_5eTycSGRj720YntldQ-CQ351RaDPc6XAU,16674
@@ -278,11 +278,12 @@ torch_geometric/llm/models/llm.py,sha256=LKYdUSw2PSWYEN13DYYaJt0dHRQTptX6mM1rAh7
 torch_geometric/llm/models/llm_judge.py,sha256=qhc8hmIPNhcfLVRyBVk7jQW7ncoIb9QYw7rcsGAIpyg,6457
 torch_geometric/llm/models/molecule_gpt.py,sha256=RWoP4RMsoRzZtuedPCLNCfooqibCqxkuAhH-pyek9No,7641
 torch_geometric/llm/models/protein_mpnn.py,sha256=SwTgafSbI2KJ-yqzn0trZtVWLmfo0_kPEaWSNJUCt70,12266
-torch_geometric/llm/models/sentence_transformer.py,sha256=vubAluqfxVKevuPpxl0c43ps_f3N0ZMi1_cRIV9EwuA,6076
+torch_geometric/llm/models/sentence_transformer.py,sha256=TSXBxeTktj10YU-h_1prdMCCmAG8MTOKMm760ch4g30,6377
 torch_geometric/llm/models/txt2kg.py,sha256=CjWXCa_WrqIlBfOSPMAK-mnaX19dvH_YhXMWSuLkk4o,14074
-torch_geometric/llm/models/vision_transformer.py,sha256=diVBefjIynzYs8WBlcpTeSVnw1PUecHY--B9Yd-W2hA,863
+torch_geometric/llm/models/vision_transformer.py,sha256=aPuVfpSwGR96KaicRYut49g6ShrCklbouaLwyPuwhBQ,1022
+torch_geometric/llm/utils/__init__.py,sha256=P5By_n15MqkUU1tfh87PGE--J7RVygPeDSBOTy_VlZ0,292
 torch_geometric/llm/utils/backend_utils.py,sha256=AcvZ8ym3UKCWwWoMl8sNRnj4FSl0_srHw9yxZAKiyeU,15840
-torch_geometric/llm/utils/feature_store.py,sha256=CMmDveFxLGfCUdyo0c0sygLLplZlIj-Fwxuk1kKxpQs,5885
+torch_geometric/llm/utils/feature_store.py,sha256=d60n3TlclEhlqoDEHKmvvGI6t8r0nur1BNwXyqqtj24,5903
 torch_geometric/llm/utils/graph_store.py,sha256=_Hh0aGnokUn0zvOC80xUfT4TtX_7G4KIDoEBkNXkgHY,7103
 torch_geometric/llm/utils/vectorrag.py,sha256=m51drMNXsGQTN6qkbR8QiHb8jvcrBlZDHzEjsEmtnes,4753
 torch_geometric/loader/__init__.py,sha256=w9LSTbyrLRkyrLXi_10d80csWgfKOKDRQDJXRdcfD0M,1835
@@ -311,9 +312,9 @@ torch_geometric/loader/utils.py,sha256=3hzKzIgB52QIZu7Jdn4JeXZaegIJinIQfIUP9DrUW
 torch_geometric/loader/zip_loader.py,sha256=3lt10fD15Rxm1WhWzypswGzCEwUz4h8OLCD1nE15yNg,3843
 torch_geometric/metrics/__init__.py,sha256=3krvDobW6vV5yHTjq2S2pmOXxNfysNG26muq7z48e94,699
 torch_geometric/metrics/link_pred.py,sha256=1_hE3KiRqAdZLI6QuUbjgyFC__mTyFu_RimM3bD8wRw,31678
-torch_geometric/nn/__init__.py,sha256=RrWRzEoqtR3lsO2lAzYXboLPb3uYEX2z3tLxiBIVWjc,847
+torch_geometric/nn/__init__.py,sha256=tTEKDy4vpjPNKyG1Vg9GIx7dVFJuQtBoh2M19ascGpo,880
 torch_geometric/nn/data_parallel.py,sha256=YiybTWoSFyfSzlXAamZ_-y1f7B6tvDEFHOuy_AyJz9Q,4761
-torch_geometric/nn/encoding.py,sha256=3DCOCO-XFt-lMb97sHWGN-4KeGUFY5lVo9P00SzrCNk,3559
+torch_geometric/nn/encoding.py,sha256=82fpwyOx0-STFSAJ5AzG0p2WFC9u1M4KgmKIql8hSLc,3634
 torch_geometric/nn/fx.py,sha256=PDtaHJAgodh4xf8FNl4fVxPGZJDbRaq3Q9z8qb1DNNI,16063
 torch_geometric/nn/glob.py,sha256=MdHjcUlHmFmTevzwND1_x7dXXJPzIDTBJRGOrGdZ8dQ,1088
 torch_geometric/nn/inits.py,sha256=_8FqacCLPz5Ft2zB5s6dtKGTKWtfrLyCLLuv1QvyKjk,2457
@@ -351,7 +352,7 @@ torch_geometric/nn/aggr/set_transformer.py,sha256=FG7_JizpFX14M6VSCwLSjYXYdJ1ZiQ
 torch_geometric/nn/aggr/sort.py,sha256=bvOOWnFkNOBOZih4rqVZQsjfeDX3vmXo1bpPSFD846w,2507
 torch_geometric/nn/aggr/utils.py,sha256=SQvdc0g6p_E2j0prA14MW2ekjEDvV-g545N0Q85uc-o,8625
 torch_geometric/nn/aggr/variance_preserving.py,sha256=fu-U_aGYpVLpgSFvVg0ONMe6nqoyv8tZ6Y35qMYTf9w,1126
-torch_geometric/nn/attention/__init__.py,sha256=w-jDQFpVqARJKjttTgKkD9kkAqRJl4MpASCfiNYIfr0,263
+torch_geometric/nn/attention/__init__.py,sha256=smNHoLd_JsPB83BOS9SKS9AQm17xAWbpi2NOoAhnvL0,273
 torch_geometric/nn/attention/performer.py,sha256=2PCDn4_-oNTao2-DkXIaoi18anP01OxRELF2pvp-jk8,7357
 torch_geometric/nn/attention/polynormer.py,sha256=uBxGs0nldp6oGlByqbxgEk23VeXLEd6B3myS5BOKDRs,3998
 torch_geometric/nn/attention/qformer.py,sha256=7J-pWm_vpumK38IC-iCBz4oqL-BEIofEIxJ0wfjWq9A,2338
@@ -653,7 +654,7 @@ torch_geometric/utils/undirected.py,sha256=H_nfpI0_WluOG6VfjPyldvcjL4w5USAKWu2x5
 torch_geometric/visualization/__init__.py,sha256=b-HnVesXjyJ_L1N-DnjiRiRVf7lhwKaBQF_2i5YMVSU,208
 torch_geometric/visualization/graph.py,sha256=mfZHXYfiU-CWMtfawYc80IxVwVmtK9hbIkSKhM_j7oI,14311
 torch_geometric/visualization/influence.py,sha256=CWMvuNA_Nf1sfbJmQgn58yS4OFpeKXeZPe7kEuvkUBw,477
-pyg_nightly-2.7.0.dev20250909.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
-pyg_nightly-2.7.0.dev20250909.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-pyg_nightly-2.7.0.dev20250909.dist-info/METADATA,sha256=6QdLMCEpPBFYlZGAG6D4gclmcLsKpWxCGDQ1j2CccR4,64145
-pyg_nightly-2.7.0.dev20250909.dist-info/RECORD,,
+pyg_nightly-2.7.0.dev20250910.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
+pyg_nightly-2.7.0.dev20250910.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+pyg_nightly-2.7.0.dev20250910.dist-info/METADATA,sha256=uuM-Hir5cz3xHjKnxcV5W027kHC71PY4M6eMzMVTlwg,64145
+pyg_nightly-2.7.0.dev20250910.dist-info/RECORD,,

torch_geometric/__init__.py

@@ -31,7 +31,7 @@ from .lazy_loader import LazyLoader
 contrib = LazyLoader('contrib', globals(), 'torch_geometric.contrib')
 graphgym = LazyLoader('graphgym', globals(), 'torch_geometric.graphgym')
 
-__version__ = '2.7.0.dev20250909'
+__version__ = '2.7.0.dev20250910'
 
 __all__ = [
     'Index',

torch_geometric/llm/__init__.py

@@ -4,6 +4,6 @@ from .utils import *  # noqa
 from .models import *  # noqa
 
 __all__ = classes = [
-    LargeGraphIndexer,
-    RAGQueryLoader,
+    'LargeGraphIndexer',
+    'RAGQueryLoader',
 ]

torch_geometric/llm/models/__init__.py

@@ -9,7 +9,7 @@ from .glem import GLEM
 from .protein_mpnn import ProteinMPNN
 from .git_mol import GITMol
 
-__all__ = [
+__all__ = classes = [
     'SentenceTransformer',
     'VisionTransformer',
     'LLM',

torch_geometric/llm/models/sentence_transformer.py

@@ -15,6 +15,13 @@ class PoolingStrategy(Enum):
 
 
 class SentenceTransformer(torch.nn.Module):
+    r"""A wrapper around a Sentence-Transformer from HuggingFace.
+
+    Args:
+        model_name (str): The HuggingFace model name, *e.g.*, :obj:`"BERT"`.
+        pooling_strategy (str, optional): The pooling strategy to use
+            for generating node embeddings. (default: :obj:`"mean"`)
+    """
     def __init__(
         self,
         model_name: str,

torch_geometric/llm/models/vision_transformer.py

@@ -5,6 +5,11 @@ from torch import Tensor
 
 
 class VisionTransformer(torch.nn.Module):
+    r"""A wrapper around a Vision-Transformer from HuggingFace.
+
+    Args:
+        model_name (str): The HuggingFace model name, *e.g.*, :obj:`"ViT"`.
+    """
     def __init__(
         self,
         model_name: str,

torch_geometric/llm/utils/__init__.py

@@ -0,0 +1,10 @@
+from .backend_utils import *  # noqa
+from .feature_store import KNNRAGFeatureStore
+from .graph_store import NeighborSamplingRAGGraphStore
+from .vectorrag import DocumentRetriever
+
+__all__ = classes = [
+    'KNNRAGFeatureStore',
+    'NeighborSamplingRAGGraphStore',
+    'DocumentRetriever',
+]

torch_geometric/llm/utils/feature_store.py

@@ -79,11 +79,11 @@ class KNNRAGFeatureStore(LocalFeatureStore):
         """Retrieves the k_nodes most similar nodes to the given query.
 
         Args:
-        - query (Union[str, List[str], Tuple[str]]):
-            The query or list of queries to search for.
+            query (Union[str, List[str], Tuple[str]]): The query
+                or list of queries to search for.
 
         Returns:
-        - The indices of the most similar nodes and the encoded query
+            The indices of the most similar nodes and the encoded query
         """
         if not isinstance(query, (list, tuple)):
             query = [query]
@@ -130,12 +130,12 @@ class KNNRAGFeatureStore(LocalFeatureStore):
         """Loads a subgraph from the given sample.
 
         Args:
-        - sample: The sample to load the subgraph from.
-        - induced: Whether to return the induced subgraph.
-            Resets node and edge ids.
+            sample: The sample to load the subgraph from.
+            induced: Whether to return the induced subgraph.
+                Resets node and edge ids.
 
         Returns:
-        - The loaded subgraph.
+            The loaded subgraph.
         """
         if isinstance(sample, HeteroSamplerOutput):
             raise NotImplementedError

torch_geometric/nn/__init__.py

@@ -8,6 +8,7 @@ from .encoding import PositionalEncoding, TemporalEncoding
 from .summary import summary
 
 from .aggr import *  # noqa
+from .attention import *  # noqa
 from .conv import *  # noqa
 from .pool import *  # noqa
 from .glob import *  # noqa

torch_geometric/nn/attention/__init__.py

@@ -3,7 +3,7 @@ from .qformer import QFormer
 from .sgformer import SGFormerAttention
 from .polynormer import PolynormerAttention
 
-__all__ = [
+__all__ = classes = [
     'PerformerAttention',
     'QFormer',
     'SGFormerAttention',

torch_geometric/nn/encoding.py

@@ -4,6 +4,11 @@ from typing import Optional
 import torch
 from torch import Tensor
 
+__all__ = classes = [
+    'PositionalEncoding',
+    'TemporalEncoding',
+]
+
 
 class PositionalEncoding(torch.nn.Module):
     r"""The positional encoding scheme from the `"Attention Is All You Need"