pyg-nightly 2.7.0.dev20250218__py3-none-any.whl → 2.7.0.dev20250220__py3-none-any.whl

{pyg_nightly-2.7.0.dev20250218.dist-info → pyg_nightly-2.7.0.dev20250220.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: pyg-nightly
-Version: 2.7.0.dev20250218
+Version: 2.7.0.dev20250220
 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>
@@ -14,6 +14,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3 :: Only
+License-File: LICENSE
 Requires-Dist: aiohttp
 Requires-Dist: fsspec
 Requires-Dist: jinja2

{pyg_nightly-2.7.0.dev20250218.dist-info → pyg_nightly-2.7.0.dev20250220.dist-info}/RECORD

@@ -1,4 +1,4 @@
-torch_geometric/__init__.py,sha256=Nr_BRj3gcIVodpmFtDtfRYq9XcwVaPTbw4R6vQGfl_Y,1978
+torch_geometric/__init__.py,sha256=Ht6O12SaZRLjd7aAphjjLY_T59xQTnRQ9nN1oSA1Xmk,1978
 torch_geometric/_compile.py,sha256=f-WQeH4VLi5Hn9lrgztFUCSrN_FImjhQa6BxFzcYC38,1338
 torch_geometric/_onnx.py,sha256=V9ffrIKSqhDw6xUZ12lkuSfNs48cQp2EeJ6Z19GfnVw,349
 torch_geometric/backend.py,sha256=lVaf7aLoVaB3M-UcByUJ1G4T4FOK6LXAg0CF4W3E8jo,1575
@@ -9,7 +9,7 @@ torch_geometric/deprecation.py,sha256=dWRymDIUkUVI2MeEmBG5WF4R6jObZeseSBV9G6FNfj
 torch_geometric/device.py,sha256=tU5-_lBNVbVHl_kUmWPwiG5mQ1pyapwMF4JkmtNN3MM,1224
 torch_geometric/edge_index.py,sha256=BsLh5tOZRjjSYDkjqOFAdBuvMaDg7EWaaLELYsUL0Z8,70048
 torch_geometric/experimental.py,sha256=JbtNNEXjFGI8hZ9raM6-qrZURP6Z5nlDK8QicZUIbz0,4756
-torch_geometric/hash_tensor.py,sha256=w4X3vI0vq0EiEn7NUrhr7pRf8p3Tn6PgxTFdjWZ8Hzw,8964
+torch_geometric/hash_tensor.py,sha256=T4fRmS6TD-j7PKa9LMralSbI6naqA4ctW8AnjWsREzw,9615
 torch_geometric/home.py,sha256=EV54B4Dmiv61GDbkCwtCfWGWJ4eFGwZ8s3KOgGjwYgY,790
 torch_geometric/index.py,sha256=9ChzWFCwj2slNcVBOgfV-wQn-KscJe_y7502w-Vf76w,24045
 torch_geometric/inspector.py,sha256=nKi5o4Mn6xsG0Ex1GudTEQt_EqnF9mcMqGtp7Shh9sQ,19336
@@ -291,7 +291,7 @@ torch_geometric/loader/utils.py,sha256=f27mczQ7fEP2HpTsJGJxKS0slPu0j8zTba3jP8ViN
 torch_geometric/loader/zip_loader.py,sha256=3lt10fD15Rxm1WhWzypswGzCEwUz4h8OLCD1nE15yNg,3843
 torch_geometric/metrics/__init__.py,sha256=3krvDobW6vV5yHTjq2S2pmOXxNfysNG26muq7z48e94,699
 torch_geometric/metrics/link_pred.py,sha256=cz9GbvZthV2PAnVnxiZlksGr0VmTQOJGNuZ-OYYg04U,29667
-torch_geometric/nn/__init__.py,sha256=RrWRzEoqtR3lsO2lAzYXboLPb3uYEX2z3tLxiBIVWjc,847
+torch_geometric/nn/__init__.py,sha256=kQHHHUxFDht2ztD-XFQuv98TvC8MdodaFsIjAvltJBw,874
 torch_geometric/nn/data_parallel.py,sha256=lDAxRi83UNuzAQSj3eu9K2sQheOIU6wqR5elS6oDs90,4764
 torch_geometric/nn/encoding.py,sha256=QNjwWczYExZ1wRGBmpuqYbn6tB7NC4BU-DEgzjhcZqw,3115
 torch_geometric/nn/fx.py,sha256=oRfnYiih0FM1MhPNcDYIog0oQ0G0soQJuaz7KeNCOjo,16048
@@ -633,7 +633,7 @@ 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.dev20250218.dist-info/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
-pyg_nightly-2.7.0.dev20250218.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
-pyg_nightly-2.7.0.dev20250218.dist-info/METADATA,sha256=SOvMlvlrvnaA_a_9QjRp6ZYd-33-2itewBSZwWu-Jzw,62999
-pyg_nightly-2.7.0.dev20250218.dist-info/RECORD,,
+pyg_nightly-2.7.0.dev20250220.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
+pyg_nightly-2.7.0.dev20250220.dist-info/WHEEL,sha256=_2ozNFCLWc93bK4WKHCO-eDUENDlo-dgc9cU3qokYO4,82
+pyg_nightly-2.7.0.dev20250220.dist-info/METADATA,sha256=SsXACQ7At6aou2g3LkqAlRX2zcjw3aTPw-mMliTKU1E,63021
+pyg_nightly-2.7.0.dev20250220.dist-info/RECORD,,

{pyg_nightly-2.7.0.dev20250218.dist-info → pyg_nightly-2.7.0.dev20250220.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: flit 3.10.1
+Generator: flit 3.11.0
 Root-Is-Purelib: true
 Tag: py3-none-any

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.dev20250218'
+__version__ = '2.7.0.dev20250220'
 
 __all__ = [
     'Index',

torch_geometric/hash_tensor.py

@@ -197,6 +197,8 @@ class HashTensor(Tensor):
 
         return out
 
+    # Methods #################################################################
+
     def as_tensor(self) -> Tensor:
         r"""Zero-copies the :class:`HashTensor` representation back to a
         :class:`torch.Tensor` representation.
@@ -205,6 +207,8 @@ class HashTensor(Tensor):
             return self._value
         return torch.arange(self.size(0), dtype=self.dtype, device=self.device)
 
+    # PyTorch/Python builtins #################################################
+
     # Prevent auto-wrapping outputs back into the proper subclass type:
     __torch_function__ = torch._C._disabled_torch_function_impl
 
@@ -267,7 +271,7 @@ def _to_copy(
         key = aten._to_copy.default(key, device=device)
         _map = get_hash_map(key)
 
-    return tensor.__class__._from_data(
+    return tensor._from_data(
         _map,
         value,
         min_key,
@@ -275,3 +279,19 @@ def _to_copy(
         num_keys=tensor.size(0),
         dtype=dtype or tensor.dtype,
     )
+
+
+@implements(aten.unsqueeze.default)
+def _unsqueeze(tensor: HashTensor, dim: int) -> HashTensor:
+    if dim == 0 or dim == -(tensor.dim() + 1):
+        raise IndexError(f"Cannot unsqueeze '{tensor.__class__.__name__}' in "
+                         f"the first dimension")
+
+    return tensor._from_data(
+        tensor._map,
+        aten.unsqueeze.default(tensor.as_tensor(), dim),
+        tensor._min_key,
+        tensor._max_key,
+        num_keys=tensor.size(0),
+        dtype=tensor.dtype,
+    )

torch_geometric/nn/__init__.py

@@ -17,6 +17,7 @@ from .dense import *  # noqa
 from .kge import *  # noqa
 from .models import *  # noqa
 from .functional import *  # noqa
+from .nlp import *  # noqa
 
 __all__ = [
     'Reshape',