pyg-nightly 2.7.0.dev20250120__py3-none-any.whl → 2.7.0.dev20250122__py3-none-any.whl

{pyg_nightly-2.7.0.dev20250120.dist-info → pyg_nightly-2.7.0.dev20250122.dist-info}/METADATA

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

@@ -1,4 +1,4 @@
-torch_geometric/__init__.py,sha256=jrg5LNbC0DH4zTFDIXy-xehMH1y8WvDTmO-a47_hlGY,1904
+torch_geometric/__init__.py,sha256=6uAYPnzs2ioPbhZa8-9MVViVgklvCD1v_1Vc8B9t3zI,1904
 torch_geometric/_compile.py,sha256=f-WQeH4VLi5Hn9lrgztFUCSrN_FImjhQa6BxFzcYC38,1338
 torch_geometric/_onnx.py,sha256=V9ffrIKSqhDw6xUZ12lkuSfNs48cQp2EeJ6Z19GfnVw,349
 torch_geometric/backend.py,sha256=lVaf7aLoVaB3M-UcByUJ1G4T4FOK6LXAg0CF4W3E8jo,1575
@@ -287,8 +287,8 @@ torch_geometric/loader/shadow.py,sha256=_hCspYf9SlJYX0lqEjxFec9e9t1iMScNThOoWR1w
 torch_geometric/loader/temporal_dataloader.py,sha256=AQ2QFeiXKbPp6I8sUeE8H7br-1_yndivXt7Z6_w62zI,2248
 torch_geometric/loader/utils.py,sha256=f27mczQ7fEP2HpTsJGJxKS0slPu0j8zTba3jP8ViNck,14901
 torch_geometric/loader/zip_loader.py,sha256=3lt10fD15Rxm1WhWzypswGzCEwUz4h8OLCD1nE15yNg,3843
-torch_geometric/metrics/__init__.py,sha256=xHDTWEG4kdv9xb5pGPlRfQjC5P-ZGbhJ0xDe3YNq3ss,393
-torch_geometric/metrics/link_pred.py,sha256=8H74eS28AcGtOUB0g8_xUUp8IX-zmlIFuBURp7Dx0No,18269
+torch_geometric/metrics/__init__.py,sha256=Ck-rqQOGoKYRNSI95zlsTtQKVN4ZJGDaYYTMeSOjvvU,435
+torch_geometric/metrics/link_pred.py,sha256=dEsKVrjJ3OYnDbeDxowMNTCQMjKdgj-Xsh0L2VY5hLo,18563
 torch_geometric/nn/__init__.py,sha256=RrWRzEoqtR3lsO2lAzYXboLPb3uYEX2z3tLxiBIVWjc,847
 torch_geometric/nn/data_parallel.py,sha256=lDAxRi83UNuzAQSj3eu9K2sQheOIU6wqR5elS6oDs90,4764
 torch_geometric/nn/encoding.py,sha256=QNjwWczYExZ1wRGBmpuqYbn6tB7NC4BU-DEgzjhcZqw,3115
@@ -459,7 +459,7 @@ torch_geometric/nn/models/signed_gcn.py,sha256=J40CnedFIqtKI1LhW1ITSEFRbA_XiJZL6
 torch_geometric/nn/models/tgn.py,sha256=kEGdfLJybkbMT4UMoAh2nCzfX3_nDjfm1cicuPHEwAM,11878
 torch_geometric/nn/models/visnet.py,sha256=97OFMCsPDEI5BCSi7RhoRcU2CNRp7zck2tEzrltFZj4,43192
 torch_geometric/nn/nlp/__init__.py,sha256=q6CPUiJHcc9bXw90lyj-ID4F3kfW8uPM-SOxW9uCMHs,213
-torch_geometric/nn/nlp/llm.py,sha256=vcFvqW-veEfVZDLSHKFKXY-1k0TbiOzmf3LZIwIA0zM,12146
+torch_geometric/nn/nlp/llm.py,sha256=j03tyCO1ADgrzGhLqYOUcsy0haGbV4dmT9bdwEnESPE,12181
 torch_geometric/nn/nlp/sentence_transformer.py,sha256=q5M7SGtrUzoSiNhKCGFb7JatWiukdhNF6zdq2yiqxwE,4475
 torch_geometric/nn/nlp/vision_transformer.py,sha256=diVBefjIynzYs8WBlcpTeSVnw1PUecHY--B9Yd-W2hA,863
 torch_geometric/nn/norm/__init__.py,sha256=u2qIDrkbeuObGVXSAIftAlvSd6ouGTtxznCfD-59UiA,669
@@ -629,6 +629,6 @@ torch_geometric/utils/undirected.py,sha256=H_nfpI0_WluOG6VfjPyldvcjL4w5USAKWu2x5
 torch_geometric/visualization/__init__.py,sha256=PyR_4K5SafsJrBr6qWrkjKr6GBL1b7FtZybyXCDEVwY,154
 torch_geometric/visualization/graph.py,sha256=ZuLPL92yGRi7lxlqsUPwL_EVVXF7P2kMcveTtW79vpA,4784
 torch_geometric/visualization/influence.py,sha256=CWMvuNA_Nf1sfbJmQgn58yS4OFpeKXeZPe7kEuvkUBw,477
-pyg_nightly-2.7.0.dev20250120.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
-pyg_nightly-2.7.0.dev20250120.dist-info/METADATA,sha256=v8PrbeT1NhlxaPY2XluG7JhiGCSKagoAogZ5NCC_6B8,62977
-pyg_nightly-2.7.0.dev20250120.dist-info/RECORD,,
+pyg_nightly-2.7.0.dev20250122.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
+pyg_nightly-2.7.0.dev20250122.dist-info/METADATA,sha256=JOKsGMFgqMMCa0rxpsjhLPD5bqoJ1lXrzBALYWKHWYk,62977
+pyg_nightly-2.7.0.dev20250122.dist-info/RECORD,,

torch_geometric/__init__.py

@@ -30,7 +30,7 @@ from .lazy_loader import LazyLoader
 contrib = LazyLoader('contrib', globals(), 'torch_geometric.contrib')
 graphgym = LazyLoader('graphgym', globals(), 'torch_geometric.graphgym')
 
-__version__ = '2.7.0.dev20250120'
+__version__ = '2.7.0.dev20250122'
 
 __all__ = [
     'Index',

torch_geometric/metrics/__init__.py

@@ -1,6 +1,7 @@
 # flake8: noqa
 
 from .link_pred import (
+    LinkPredMetric,
     LinkPredMetricCollection,
     LinkPredPrecision,
     LinkPredRecall,
@@ -11,6 +12,7 @@ from .link_pred import (
 )
 
 link_pred_metrics = [
+    'LinkPredMetric',
     'LinkPredMetricCollection',
     'LinkPredPrecision',
     'LinkPredRecall',

torch_geometric/metrics/link_pred.py

@@ -30,6 +30,14 @@ class LinkPredMetricData:
         if hasattr(self, '_pred_rel_mat'):
             return self._pred_rel_mat  # type: ignore
 
+        if self.edge_label_index[1].numel() == 0:
+            self._pred_rel_mat = torch.zeros_like(
+                self.pred_index_mat,
+                dtype=torch.bool if self.edge_label_weight is None else
+                torch.get_default_dtype(),
+            )
+            return self._pred_rel_mat
+
         # Flatten both prediction and ground-truth indices, and determine
         # overlaps afterwards via `torch.searchsorted`.
         max_index = max(  # type: ignore
@@ -139,7 +147,7 @@ class LinkPredMetric(BaseMetric):
     is_differentiable: bool = False
     full_state_update: bool = False
     higher_is_better: Optional[bool] = None
-    weighted: bool = False
+    weighted: bool
 
     def __init__(self, k: int) -> None:
         super().__init__()
@@ -484,7 +492,7 @@ class LinkPredNDCG(LinkPredMetric):
                 self.discount,
                 self.discount.new_full((1, ), fill_value=float('inf')),
             ])
-            discount = discount[pos.clamp(max=self.k + 1)]
+            discount = discount[pos.clamp(max=self.k)]
 
             idcg = scatter(  # Apply discount and aggregate:
                 data.edge_label_weight / discount,

torch_geometric/nn/nlp/llm.py

@@ -62,8 +62,8 @@ class LLM(torch.nn.Module):
     def __init__(
         self,
         model_name: str,
-        num_params: int = None,
-        dtype=torch.bfloat16,
+        num_params: Optional[int] = None,
+        dtype: Optional[torch.dtype] = torch.bfloat16,
     ) -> None:
         super().__init__()