pyg-nightly 2.7.0.dev20250920__py3-none-any.whl → 2.7.0.dev20250921__py3-none-any.whl

{pyg_nightly-2.7.0.dev20250920.dist-info → pyg_nightly-2.7.0.dev20250921.dist-info}/METADATA

@@ -1,15 +1,14 @@
 Metadata-Version: 2.4
 Name: pyg-nightly
-Version: 2.7.0.dev20250920
+Version: 2.7.0.dev20250921
 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>
-Requires-Python: >=3.9
+Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-Expression: MIT
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Programming Language :: Python
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
@@ -45,7 +44,6 @@ Requires-Dist: networkx ; extra == "full"
 Requires-Dist: numba<0.60.0 ; extra == "full"
 Requires-Dist: opt_einsum ; extra == "full"
 Requires-Dist: pandas ; extra == "full"
-Requires-Dist: pgmpy ; extra == "full"
 Requires-Dist: pynndescent ; extra == "full"
 Requires-Dist: pytorch-memlab ; extra == "full"
 Requires-Dist: rdflib ; extra == "full"
@@ -91,14 +89,21 @@ Provides-Extra: test
 
 ______________________________________________________________________
 
+<div align="center">
+
 [![PyPI Version][pypi-image]][pypi-url]
-[![Testing Status][testing-image]][testing-url]
-[![Linting Status][linting-image]][linting-url]
-[![Docs Status][docs-image]][docs-url]
-[![Contributing][contributing-image]][contributing-url]
+[![PyPI Download][pypi-download-image]][pypi-download-url]
 [![Slack][slack-image]][slack-url]
+[![Contributing][contributing-image]][contributing-url]
+
+**[Documentation](https://pytorch-geometric.readthedocs.io)** |
+**[PyG 1.0 Paper](https://arxiv.org/abs/1903.02428)** |
+**[PyG 2.0 Paper](https://arxiv.org/abs/2507.16991)** |
+**[Colab Notebooks](https://pytorch-geometric.readthedocs.io/en/latest/get_started/colabs.html)** |
+**[External Resources](https://pytorch-geometric.readthedocs.io/en/latest/external/resources.html)** |
+**[OGB Examples](https://github.com/snap-stanford/ogb/tree/master/examples)**
 
-**[Documentation](https://pytorch-geometric.readthedocs.io)** | **[PyG 1.0 Paper](https://arxiv.org/abs/1903.02428)** | **[PyG 2.0 Paper](https://arxiv.org/abs/2507.16991)** | **[Colab Notebooks](https://pytorch-geometric.readthedocs.io/en/latest/get_started/colabs.html)** | **[External Resources](https://pytorch-geometric.readthedocs.io/en/latest/external/resources.html)** | **[OGB Examples](https://github.com/snap-stanford/ogb/tree/master/examples)**
+</div>
 
 **PyG** *(PyTorch Geometric)* is a library built upon [PyTorch](https://pytorch.org/) to easily write and train Graph Neural Networks (GNNs) for a wide range of applications related to structured data.
 
@@ -421,7 +426,7 @@ These approaches have been implemented in PyG, and can benefit from the above GN
 
 ## Installation
 
-PyG is available for Python 3.9 to Python 3.13.
+PyG is available for Python 3.10 to Python 3.13.
 
 From **PyG 2.3** onwards, you can install and use PyG **without any external library** required except for PyTorch.
 For this, simply run
@@ -546,16 +551,12 @@ If you notice anything unexpected, please open an [issue](https://github.com/pyg
 If you have any questions or are missing a specific feature, feel free [to discuss them with us](https://github.com/pyg-team/pytorch_geometric/discussions).
 We are motivated to constantly make PyG even better.
 
-[contributing-image]: https://img.shields.io/badge/contributions-welcome-brightgreen.svg?style=flat
+[contributing-image]: https://img.shields.io/badge/contributions-welcome-brightgreen.svg?style=flat&color=4B26A4
 [contributing-url]: https://github.com/pyg-team/pytorch_geometric/blob/master/.github/CONTRIBUTING.md
-[docs-image]: https://readthedocs.org/projects/pytorch-geometric/badge/?version=latest
-[docs-url]: https://pytorch-geometric.readthedocs.io/en/latest
-[linting-image]: https://github.com/pyg-team/pytorch_geometric/actions/workflows/linting.yml/badge.svg
-[linting-url]: https://github.com/pyg-team/pytorch_geometric/actions/workflows/linting.yml
-[pypi-image]: https://badge.fury.io/py/torch-geometric.svg
+[pypi-download-image]: https://img.shields.io/pypi/dm/torch_geometric?color=4B26A4
+[pypi-download-url]: https://pepy.tech/projects/torch_geometric
+[pypi-image]: https://img.shields.io/pypi/pyversions/torch-geometric?color=4B26A4
 [pypi-url]: https://pypi.python.org/pypi/torch-geometric
-[slack-image]: https://img.shields.io/badge/slack-pyg-brightgreen
+[slack-image]: https://img.shields.io/badge/slack-join-white.svg?logo=slack&color=4B26A4
 [slack-url]: https://data.pyg.org/slack.html
-[testing-image]: https://github.com/pyg-team/pytorch_geometric/actions/workflows/testing.yml/badge.svg
-[testing-url]: https://github.com/pyg-team/pytorch_geometric/actions/workflows/testing.yml
 

{pyg_nightly-2.7.0.dev20250920.dist-info → pyg_nightly-2.7.0.dev20250921.dist-info}/RECORD

@@ -1,9 +1,9 @@
-torch_geometric/__init__.py,sha256=0ij8VxVSK4T5A19Dr05CrBi0vbfTv2d0vTpB73hQsws,2292
+torch_geometric/__init__.py,sha256=pOD9vZjRGqoUzFw5_-UNQVsobbiDjoaicP9Jw5mj9jE,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
 torch_geometric/config_mixin.py,sha256=hOTJu5LLVrEAZ6Pjt4ScLDLKv9aHbfAzF_3ufwKgO4I,4301
-torch_geometric/config_store.py,sha256=zdMzlgBpUmBkPovpYQh5fMNwTZLDq2OneqX47QEx7zk,16818
+torch_geometric/config_store.py,sha256=Lj5pY_fFamF6pr5XhaKcYXZ3ecUTFNHWEprnBwaNbso,16801
 torch_geometric/debug.py,sha256=cLyH9OaL2v7POyW-80b19w-ctA7a_5EZsS4aUF1wc2U,1295
 torch_geometric/deprecation.py,sha256=gN65uX23c3miRPOpQzxcRS_QDUpD3B-qVKD6y6GX8Yw,872
 torch_geometric/device.py,sha256=tU5-_lBNVbVHl_kUmWPwiG5mQ1pyapwMF4JkmtNN3MM,1224
@@ -19,7 +19,7 @@ torch_geometric/logging.py,sha256=HmHHLiCcM64k-6UYNOSfXPIeSGNAyiGGcn8cD8tlyuQ,85
 torch_geometric/resolver.py,sha256=fn-_6mCpI2xv7eDZnIFcYrHOn0IrwbkWFLDb9laQrWI,1270
 torch_geometric/seed.py,sha256=MJLbVwpb9i8mK3oi32sS__Cq-dRq_afTeoOL_HoA9ko,372
 torch_geometric/template.py,sha256=rqjDWgcSAgTCiV4bkOjWRPaO4PpUdC_RXigzxxBqAu8,1060
-torch_geometric/typing.py,sha256=PenHvJZ2ZQ0V6t5hYypcZq8mQ47upgSlVTqHPy6QvLc,15704
+torch_geometric/typing.py,sha256=Kj0R-aw81aY6JE1pbsG_7Jf830tect31uzibDFsjSPc,15684
 torch_geometric/warnings.py,sha256=SB9dWGovX_KKcxqsOrdTDvSb_j0NoB5vPGnK2vg0jVw,727
 torch_geometric/contrib/__init__.py,sha256=To_lDofgM7sogKEYOICIXei7Njuk-Vkfm-OFhPIdaLo,366
 torch_geometric/contrib/datasets/__init__.py,sha256=lrGnWsEiJf5zsBRmshGZZFN_uYR2ezDjbj9n9nCpvtk,23
@@ -36,7 +36,7 @@ torch_geometric/data/collate.py,sha256=tOUvttXoEo-bOvJx_qMivJq2JqOsB9iDdjovtiyys
 torch_geometric/data/data.py,sha256=-E6el1knNgSJyapV8KUk2aRRHOfvwEvjUFfe_BapLfc,47490
 torch_geometric/data/database.py,sha256=K3KLefYVfsBN9HRItgFZNkbUIllfDt4ueauBFxk3Rxk,23106
 torch_geometric/data/datapipes.py,sha256=9_Cq3j_7LIF4plQFzbLaqyy0LcpKdAic6yiKgMqSX9A,3083
-torch_geometric/data/dataset.py,sha256=AaJH0N9eZgvxX0ljyTH8cXutKJ0AGFAyE-H4Sw9D51w,16834
+torch_geometric/data/dataset.py,sha256=-B9lXmpJ5-yhLZ-38PzKaaDeFZ1y-hfsTn-bVrnzrYA,16886
 torch_geometric/data/download.py,sha256=kcesTu6jlgmCeePpOxDQOnVhxB_GuZ9iu9ds72KEORc,1889
 torch_geometric/data/extract.py,sha256=DMG8_6ps4O5xKfkb7j1gUBX_jlWpFdmz6OLY2jBSEx4,2339
 torch_geometric/data/feature_store.py,sha256=pl2pJL25wqzEZnNZbW8c8Ee_yH0DnE2AK8TioTWZV-g,20045
@@ -118,7 +118,7 @@ torch_geometric/datasets/medshapenet.py,sha256=eCBCXKpueweCwDSf_Q4_MwVA3IbJd04FS
 torch_geometric/datasets/mixhop_synthetic_dataset.py,sha256=4NNvTHUvvV6pcqQCyVDS5XhppXUeF2H9GTfFoc49eyU,3951
 torch_geometric/datasets/mnist_superpixels.py,sha256=o2ArbZ0_OE0u8VCaHmWwvngESlOFr9oM9dSEP_tjAS4,3340
 torch_geometric/datasets/modelnet.py,sha256=rqR-e75lC8PS_IX7VlNbo2Az9IWfqMNvDp8rmQCp-LE,5357
-torch_geometric/datasets/molecule_gpt_dataset.py,sha256=d6V8_Qy1Y2nt9hFhn7Re1omFw5Tf_uhv13QM0Vg76eg,19091
+torch_geometric/datasets/molecule_gpt_dataset.py,sha256=DUzX1h7a0qsIhUVzLyNb5qarxZYFrxBr4bsQJ7vsJrk,19096
 torch_geometric/datasets/molecule_net.py,sha256=pMzaJzd-LbBncZ0VoC87HfA8d1F4NwCWTb5YKvLM890,7404
 torch_geometric/datasets/movie_lens.py,sha256=M4Bu0Xus8IkW8GYzjxPxSdPXNbcCCx9cu6cncxBvLx8,4033
 torch_geometric/datasets/movie_lens_100k.py,sha256=eTpBAteM3jqTEtiwLxmhVj4r8JvftvPx8Hvs-3ZIHlU,6057
@@ -181,7 +181,7 @@ torch_geometric/datasets/motif_generator/grid.py,sha256=Pcv3r80nfzqpBvRTAJT_J0Dp
 torch_geometric/datasets/motif_generator/house.py,sha256=C_E2EgeqXEB9CHKRd9V5Jji4lFPpJb_3c41vYSk9Gcs,814
 torch_geometric/datasets/utils/__init__.py,sha256=At_dId4MdpzAkDS_7Mc6I7XlkThbL0AbVzHC_92lcjA,182
 torch_geometric/datasets/utils/cheatsheet.py,sha256=M55Bj64cjMVqDNoIq1shUVeU2ngoxpEjhdtyqw7Sd_k,1835
-torch_geometric/distributed/__init__.py,sha256=NNCGXbDTAW5xoJgSr-PK0VYEnT8UCI7SoZXc16fjuxQ,589
+torch_geometric/distributed/__init__.py,sha256=FUsRoS28c7XfnwlA9yF4m2xZWos5TdpQSuY3Bm48vz8,1108
 torch_geometric/distributed/dist_context.py,sha256=n34e2HU-TxmK6DrOpb5lWZu_xg1To1IFrXH4ueF_Jhg,418
 torch_geometric/distributed/dist_link_neighbor_loader.py,sha256=wM9heZmStrPSW7eo9qWusKdI_lVkDkLlda8ILBqC2c8,4933
 torch_geometric/distributed/dist_loader.py,sha256=Gjvl5Ck8YrFN6YmCWEFWVqLEwI1hog-rWj2Sk_zqYC0,6504
@@ -453,7 +453,7 @@ torch_geometric/nn/models/__init__.py,sha256=7eRlAR93pltDfLEcsTJaaK67mVFezAIjg0V
 torch_geometric/nn/models/attentive_fp.py,sha256=1z3iTV2O5W9tqHFAdno8FeBFeXmuG-TDZk4lwwVh3Ac,6634
 torch_geometric/nn/models/attract_repel.py,sha256=h9OyogT0NY0xiT0DkpJHMxH6ZUmo8R-CmwZdKEwq8Ek,5277
 torch_geometric/nn/models/autoencoder.py,sha256=nGje-zty78Y3hxOJ9o0_6QziJjOvBlknk6z0_fDQwQU,10770
-torch_geometric/nn/models/basic_gnn.py,sha256=PGa0RUMyvrNy_5yRI2jX_zwPsmZXwOQWfsWvxOiHsSk,31225
+torch_geometric/nn/models/basic_gnn.py,sha256=tp7qbHKn_uO1CBaEiW79zaBDAD-fR88E8ffJpdDYr9w,31261
 torch_geometric/nn/models/captum.py,sha256=vPN85_HDMTNcw-rKXAtYY-vT2SbHdf4CFtkseqYsnHg,3972
 torch_geometric/nn/models/correct_and_smooth.py,sha256=wmq-US2r4ocd0a661R8YeDiBeVtILOjdN-4swIth9BQ,6827
 torch_geometric/nn/models/deep_graph_infomax.py,sha256=yXSZ4mCrq4Dcvl1muzkxEWH4Lo525J4cYuAXpGs55IY,4137
@@ -654,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.dev20250920.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
-pyg_nightly-2.7.0.dev20250920.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-pyg_nightly-2.7.0.dev20250920.dist-info/METADATA,sha256=PAeahjszlJpaI4WHs-eZPOYELiodtDDAPudxTK4MfTA,64145
-pyg_nightly-2.7.0.dev20250920.dist-info/RECORD,,
+pyg_nightly-2.7.0.dev20250921.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
+pyg_nightly-2.7.0.dev20250921.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+pyg_nightly-2.7.0.dev20250921.dist-info/METADATA,sha256=A3uqoYW6Tuh_CSAyaqwO_WS0CY4l-r6_kQZAu_PYo_U,63680
+pyg_nightly-2.7.0.dev20250921.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.dev20250920'
+__version__ = '2.7.0.dev20250921'
 
 __all__ = [
     'Index',

torch_geometric/config_store.py

@@ -168,7 +168,7 @@ def map_annotation(
         assert origin is not None
         args = tuple(map_annotation(a, mapping) for a in args)
         if type(annotation).__name__ == 'GenericAlias':
-            # If annotated with `list[...]` or `dict[...]` (>= Python 3.10):
+            # If annotated with `list[...]` or `dict[...]`:
             annotation = origin[args]
         else:
             # If annotated with `typing.List[...]` or `typing.Dict[...]`:

torch_geometric/data/dataset.py

@@ -236,8 +236,8 @@ class Dataset(torch.utils.data.Dataset):
 
     def _process(self):
         f = osp.join(self.processed_dir, 'pre_transform.pt')
-        if osp.exists(f) and torch.load(f, weights_only=False) != _repr(
-                self.pre_transform):
+        if not self.force_reload and osp.exists(f) and torch.load(
+                f, weights_only=False) != _repr(self.pre_transform):
             warnings.warn(
                 "The `pre_transform` argument differs from the one used in "
                 "the pre-processed version of this dataset. If you want to "
@@ -246,8 +246,8 @@ class Dataset(torch.utils.data.Dataset):
                 stacklevel=2)
 
         f = osp.join(self.processed_dir, 'pre_filter.pt')
-        if osp.exists(f) and torch.load(f, weights_only=False) != _repr(
-                self.pre_filter):
+        if not self.force_reload and osp.exists(f) and torch.load(
+                f, weights_only=False) != _repr(self.pre_filter):
             warnings.warn(
                 "The `pre_filter` argument differs from the one used in "
                 "the pre-processed version of this dataset. If you want to "

torch_geometric/datasets/molecule_gpt_dataset.py

@@ -174,7 +174,7 @@ def extract_name(
 class MoleculeGPTDataset(InMemoryDataset):
     r"""The dataset from the `"MoleculeGPT: Instruction Following Large
     Language Models for Molecular Property Prediction"
-    <https://ai4d3.github.io/papers/34.pdf>`_ paper.
+    <https://ai4d3.github.io/2023/papers/34.pdf>`_ paper.
 
     Args:
         root (str): Root directory where the dataset should be saved.

torch_geometric/distributed/__init__.py

@@ -1,3 +1,5 @@
+from warnings import warn
+
 from .dist_context import DistContext
 from .local_feature_store import LocalFeatureStore
 from .local_graph_store import LocalGraphStore
@@ -7,6 +9,17 @@ from .dist_loader import DistLoader
 from .dist_neighbor_loader import DistNeighborLoader
 from .dist_link_neighbor_loader import DistLinkNeighborLoader
 
+warn(
+    "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
+    "no longer be maintained. For distributed training, refer to our "
+    "tutorials on distributed training at "
+    "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
+    "or cuGraph examples at "
+    "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
+    stacklevel=2,
+    category=DeprecationWarning,
+)
+
 __all__ = classes = [
     'DistContext',
     'LocalFeatureStore',

torch_geometric/nn/models/basic_gnn.py

@@ -415,7 +415,8 @@ class GCN(BasicGNN):
             (default: :obj:`None`)
         jk (str, optional): The Jumping Knowledge mode. If specified, the model
             will additionally apply a final linear transformation to transform
-            node embeddings to the expected output feature dimensionality.
+            node embeddings to the expected output feature dimensionality,
+            while default will not.
             (:obj:`None`, :obj:`"last"`, :obj:`"cat"`, :obj:`"max"`,
             :obj:`"lstm"`). (default: :obj:`None`)
         **kwargs (optional): Additional arguments of

torch_geometric/typing.py

@@ -3,17 +3,12 @@ import os
 import sys
 import typing
 import warnings
-from typing import Any, Dict, List, Optional, Set, Tuple, Union
+from typing import Any, Dict, List, Optional, Set, Tuple, TypeAlias, Union
 
 import numpy as np
 import torch
 from torch import Tensor
 
-try:
-    from typing import TypeAlias  # type: ignore
-except ImportError:
-    from typing_extensions import TypeAlias
-
 WITH_PT20 = int(torch.__version__.split('.')[0]) >= 2
 WITH_PT21 = WITH_PT20 and int(torch.__version__.split('.')[1]) >= 1
 WITH_PT22 = WITH_PT20 and int(torch.__version__.split('.')[1]) >= 2
@@ -98,7 +93,7 @@ except Exception as e:
     WITH_CUDA_HASH_MAP = False
 
 if WITH_CPU_HASH_MAP:
-    CPUHashMap: TypeAlias = torch.classes.pyg.CPUHashMap
+    CPUHashMap: TypeAlias = torch.classes.pyg.CPUHashMap  # type: ignore[name-defined]  # noqa: E501
 else:
 
     class CPUHashMap:  # type: ignore
@@ -110,7 +105,7 @@ else:
 
 
 if WITH_CUDA_HASH_MAP:
-    CUDAHashMap: TypeAlias = torch.classes.pyg.CUDAHashMap
+    CUDAHashMap: TypeAlias = torch.classes.pyg.CUDAHashMap  # type: ignore[name-defined]  # noqa: E501
 else:
 
     class CUDAHashMap:  # type: ignore