pyg-nightly 2.7.0.dev20250810__py3-none-any.whl → 2.7.0.dev20250812__py3-none-any.whl

{pyg_nightly-2.7.0.dev20250810.dist-info → pyg_nightly-2.7.0.dev20250812.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyg-nightly
-Version: 2.7.0.dev20250810
+Version: 2.7.0.dev20250812
 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.dev20250810.dist-info → pyg_nightly-2.7.0.dev20250812.dist-info}/RECORD

@@ -1,4 +1,4 @@
-torch_geometric/__init__.py,sha256=QVbGY_fKy5fB3YKENkALWpWLnakRDXKuL2Qtzty4LrU,2250
+torch_geometric/__init__.py,sha256=zh9Ue8-AfooFkxHFpkA1KqRe8VEybsdyAFSi7aNWXWs,2250
 torch_geometric/_compile.py,sha256=9yqMTBKatZPr40WavJz9FjNi7pQj8YZAZOyZmmRGXgc,1351
 torch_geometric/_onnx.py,sha256=V9ffrIKSqhDw6xUZ12lkuSfNs48cQp2EeJ6Z19GfnVw,349
 torch_geometric/backend.py,sha256=lVaf7aLoVaB3M-UcByUJ1G4T4FOK6LXAg0CF4W3E8jo,1575
@@ -54,7 +54,7 @@ torch_geometric/data/summary.py,sha256=o9rmAIkPEAMpsjd4UhoIWtpGsCd0oHpBS99_XkATb
 torch_geometric/data/temporal.py,sha256=WOJ6gFrTLikaLhUvotyUF5ql14FkE5Ox3hNkdSp6ZGg,10018
 torch_geometric/data/view.py,sha256=XjkVSc-UWZFCT4DlXLShZtO8duhFQkS9gq88zZXANsk,1089
 torch_geometric/data/lightning/__init__.py,sha256=w3En1tJfy3kSqe1MycpOyZpHFO3fxBCgNCUOznPA3YU,178
-torch_geometric/data/lightning/datamodule.py,sha256=IjucsIKRcNv16DIqILQnqa_sz72a4-yivoySmEllv2o,29353
+torch_geometric/data/lightning/datamodule.py,sha256=jDv9ibLQV_FyPeq8ncq77oOU8qy-STCf-aaYd0R8JE8,29545
 torch_geometric/datasets/__init__.py,sha256=rgfUmjd9U3o8renKVl81Brscx4LOtwWmt6qAoaG41C4,6417
 torch_geometric/datasets/actor.py,sha256=oUxgJIX8bi5hJr1etWNYIFyVQNDDXi1nyVpHGGMEAGQ,4304
 torch_geometric/datasets/airfrans.py,sha256=8cCBmHPttrlKY_iwfyr-K-CUX_JEDjrIOg3r9dQSN7o,5439
@@ -217,7 +217,7 @@ torch_geometric/graphgym/benchmark.py,sha256=52Y8fyMQ5Q0hS0kowBlVfnnk-pkLNQLFc8l
 torch_geometric/graphgym/checkpoint.py,sha256=v07NKxmXLn0Rqy87t0jHZtyJojRSMWdcwRRdZyWCY18,2408
 torch_geometric/graphgym/cmd_args.py,sha256=l4v_o21YYQ1uaWcKpeMz-D5_CfHKfNDQQbqatomnDgw,738
 torch_geometric/graphgym/config.py,sha256=e0TLYKlC_O3VANaYWRfKbjttFb-IhmJayOol5OiPs9g,17210
-torch_geometric/graphgym/imports.py,sha256=hPKDKra8ltQ2gXL_pxajzm7DGHDi6Ty5GkGyYlnaTnE,396
+torch_geometric/graphgym/imports.py,sha256=ymrPip1soU36szGl0YS0zs00Qf4_dG7exuCRfk8upUM,615
 torch_geometric/graphgym/init.py,sha256=qRlA1BmbtP-ve8XsD4HKpczWRnBMddtI2DdE1pHfJFQ,521
 torch_geometric/graphgym/loader.py,sha256=CEJv8Hw4kwAWr6A7f9GCoU0UIb4K4zWYDx29L7xfiJ8,11605
 torch_geometric/graphgym/logger.py,sha256=6vqMLOvxPPQcCA2o3W-jaX2PZwZt_rR66NktBwgH48Q,11272
@@ -645,7 +645,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.dev20250810.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
-pyg_nightly-2.7.0.dev20250810.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-pyg_nightly-2.7.0.dev20250810.dist-info/METADATA,sha256=tTJs9aR2sXsV0YMUWS4UZqNy_zdie9AUAGLC1cnSKyY,63484
-pyg_nightly-2.7.0.dev20250810.dist-info/RECORD,,
+pyg_nightly-2.7.0.dev20250812.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
+pyg_nightly-2.7.0.dev20250812.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+pyg_nightly-2.7.0.dev20250812.dist-info/METADATA,sha256=LjAZPBI2jftWDFxvf8tdFufw21zlKjd4GYQ_Q4LbXk4,63484
+pyg_nightly-2.7.0.dev20250812.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.dev20250810'
+__version__ = '2.7.0.dev20250812'
 
 __all__ = [
     'Index',

torch_geometric/data/lightning/datamodule.py

@@ -11,21 +11,27 @@ from torch_geometric.sampler import BaseSampler, NeighborSampler
 from torch_geometric.typing import InputEdges, InputNodes, OptTensor
 
 try:
-    from pytorch_lightning import LightningDataModule as PLLightningDataModule
-    no_pytorch_lightning = False
+    from pytorch_lightning import LightningDataModule as _LightningDataModule
+    _pl_is_available = True
 except ImportError:
-    PLLightningDataModule = object  # type: ignore
-    no_pytorch_lightning = True
+    try:
+        from lightning.pytorch import \
+            LightningDataModule as _LightningDataModule
+        _pl_is_available = True
+    except ImportError:
+        _pl_is_available = False
+        _LightningDataModule = object
 
 
-class LightningDataModule(PLLightningDataModule):
+class LightningDataModule(_LightningDataModule):
     def __init__(self, has_val: bool, has_test: bool, **kwargs: Any) -> None:
         super().__init__()
 
-        if no_pytorch_lightning:
+        if not _pl_is_available:
             raise ModuleNotFoundError(
-                "No module named 'pytorch_lightning' found on this machine. "
-                "Run 'pip install pytorch_lightning' to install the library.")
+                "No module named 'pytorch_lightning' (or 'lightning') found "
+                "in your Python environment. Run 'pip install "
+                "pytorch_lightning' or 'pip install lightning'")
 
         if not has_val:
             self.val_dataloader = None  # type: ignore

torch_geometric/graphgym/imports.py

@@ -4,14 +4,23 @@ import torch
 
 try:
     import pytorch_lightning as pl
+    _pl_is_available = True
+except ImportError:
+    try:
+        import lightning.pytorch as pl
+        _pl_is_available = True
+    except ImportError:
+        _pl_is_available = False
+
+if _pl_is_available:
     LightningModule = pl.LightningModule
     Callback = pl.Callback
-except ImportError:
+else:
     pl = object
     LightningModule = torch.nn.Module
     Callback = object
 
     warnings.warn(
-        "Please install 'pytorch_lightning' via  "
-        "'pip install pytorch_lightning' in order to use GraphGym",
+        "To use GraphGym, install 'pytorch_lightning' or 'lightning' via "
+        "'pip install pytorch_lightning' or 'pip install lightning'",
         stacklevel=2)