pyg-nightly 2.7.0.dev20250218__tar.gz → 2.7.0.dev20250220__tar.gz

{pyg_nightly-2.7.0.dev20250218 → pyg_nightly-2.7.0.dev20250220}/PKG-INFO

@@ -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 → pyg_nightly-2.7.0.dev20250220}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend="flit_core.buildapi"
 
 [project]
 name="pyg-nightly"
-version="2.7.0.dev20250218"
+version="2.7.0.dev20250220"
 authors=[
     {name="Matthias Fey", email="matthias@pyg.org"},
 ]

{pyg_nightly-2.7.0.dev20250218 → pyg_nightly-2.7.0.dev20250220}/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',

{pyg_nightly-2.7.0.dev20250218 → pyg_nightly-2.7.0.dev20250220}/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,
+    )

{pyg_nightly-2.7.0.dev20250218 → pyg_nightly-2.7.0.dev20250220}/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',