pyg-nightly 2.7.0.dev20250530__py3-none-any.whl → 2.7.0.dev20250531__py3-none-any.whl

{pyg_nightly-2.7.0.dev20250530.dist-info → pyg_nightly-2.7.0.dev20250531.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyg-nightly
-Version: 2.7.0.dev20250530
+Version: 2.7.0.dev20250531
 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>
@@ -444,39 +444,39 @@ We recommend to start with a minimal installation, and install additional depend
 
 For ease of installation of these extensions, we provide `pip` wheels for all major OS/PyTorch/CUDA combinations, see [here](https://data.pyg.org/whl).
 
-#### PyTorch 2.6
+#### PyTorch 2.7
 
-To install the binaries for PyTorch 2.6.0, simply run
+To install the binaries for PyTorch 2.7.0, simply run
 
 ```
-pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.6.0+${CUDA}.html
+pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.7.0+${CUDA}.html
 ```
 
-where `${CUDA}` should be replaced by either `cpu`, `cu118`, `cu124`, or `cu126` depending on your PyTorch installation.
+where `${CUDA}` should be replaced by either `cpu`, `cu118`, `cu126`, or `cu128` depending on your PyTorch installation.
 
-|             | `cpu` | `cu118` | `cu124` | `cu126` |
+|             | `cpu` | `cu118` | `cu126` | `cu128` |
 | ----------- | ----- | ------- | ------- | ------- |
 | **Linux**   | ✅    | ✅      | ✅      | ✅      |
 | **Windows** | ✅    | ✅      | ✅      | ✅      |
 | **macOS**   | ✅    |         |         |         |
 
-#### PyTorch 2.5
+#### PyTorch 2.6
 
-To install the binaries for PyTorch 2.5.0/2.5.1, simply run
+To install the binaries for PyTorch 2.6.0, simply run
 
 ```
-pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.5.0+${CUDA}.html
+pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.6.0+${CUDA}.html
 ```
 
-where `${CUDA}` should be replaced by either `cpu`, `cu118`, `cu121`, or `cu124` depending on your PyTorch installation.
+where `${CUDA}` should be replaced by either `cpu`, `cu118`, `cu124`, or `cu126` depending on your PyTorch installation.
 
-|             | `cpu` | `cu118` | `cu121` | `cu124` |
+|             | `cpu` | `cu118` | `cu124` | `cu126` |
 | ----------- | ----- | ------- | ------- | ------- |
 | **Linux**   | ✅    | ✅      | ✅      | ✅      |
 | **Windows** | ✅    | ✅      | ✅      | ✅      |
 | **macOS**   | ✅    |         |         |         |
 
-**Note:** Binaries of older versions are also provided for PyTorch 1.4.0, PyTorch 1.5.0, PyTorch 1.6.0, PyTorch 1.7.0/1.7.1, PyTorch 1.8.0/1.8.1, PyTorch 1.9.0, PyTorch 1.10.0/1.10.1/1.10.2, PyTorch 1.11.0, PyTorch 1.12.0/1.12.1, PyTorch 1.13.0/1.13.1, PyTorch 2.0.0/2.0.1, PyTorch 2.1.0/2.1.1/2.1.2, PyTorch 2.2.0/2.2.1/2.2.2, PyTorch 2.3.0/2.3.1, and PyTorch 2.4.0/2.4.1 (following the same procedure).
+**Note:** Binaries of older versions are also provided for PyTorch 1.4.0, PyTorch 1.5.0, PyTorch 1.6.0, PyTorch 1.7.0/1.7.1, PyTorch 1.8.0/1.8.1, PyTorch 1.9.0, PyTorch 1.10.0/1.10.1/1.10.2, PyTorch 1.11.0, PyTorch 1.12.0/1.12.1, PyTorch 1.13.0/1.13.1, PyTorch 2.0.0/2.0.1, PyTorch 2.1.0/2.1.1/2.1.2, PyTorch 2.2.0/2.2.1/2.2.2, PyTorch 2.3.0/2.3.1, PyTorch 2.4.0/2.4.1, and PyTorch 2.5.0/2.5.1 (following the same procedure).
 **For older versions, you might need to explicitly specify the latest supported version number** or install via `pip install --no-index` in order to prevent a manual installation from source.
 You can look up the latest supported version number [here](https://data.pyg.org/whl).
 

{pyg_nightly-2.7.0.dev20250530.dist-info → pyg_nightly-2.7.0.dev20250531.dist-info}/RECORD

@@ -1,4 +1,4 @@
-torch_geometric/__init__.py,sha256=t19zh1v4oBZ39p0wQeCPcE01o0D8x5_bFLx1BAIcBoY,2255
+torch_geometric/__init__.py,sha256=ATk4zdv-Np1gZlQB0z4Jxuj5My8yOgPOpXNFtYECdP8,2255
 torch_geometric/_compile.py,sha256=f-WQeH4VLi5Hn9lrgztFUCSrN_FImjhQa6BxFzcYC38,1338
 torch_geometric/_onnx.py,sha256=V9ffrIKSqhDw6xUZ12lkuSfNs48cQp2EeJ6Z19GfnVw,349
 torch_geometric/backend.py,sha256=lVaf7aLoVaB3M-UcByUJ1G4T4FOK6LXAg0CF4W3E8jo,1575
@@ -7,11 +7,11 @@ torch_geometric/config_store.py,sha256=zdMzlgBpUmBkPovpYQh5fMNwTZLDq2OneqX47QEx7
 torch_geometric/debug.py,sha256=cLyH9OaL2v7POyW-80b19w-ctA7a_5EZsS4aUF1wc2U,1295
 torch_geometric/deprecation.py,sha256=dWRymDIUkUVI2MeEmBG5WF4R6jObZeseSBV9G6FNfjc,858
 torch_geometric/device.py,sha256=tU5-_lBNVbVHl_kUmWPwiG5mQ1pyapwMF4JkmtNN3MM,1224
-torch_geometric/edge_index.py,sha256=jSWrZ77qKKydVDxiXrsBlaoL6Qdems6-HiA_B_qDo2o,70078
+torch_geometric/edge_index.py,sha256=RbIwLhtoLXmkQ_DqThBCwi1JH7zNRTsuVj0X-sTYlWE,70094
 torch_geometric/experimental.py,sha256=JbtNNEXjFGI8hZ9raM6-qrZURP6Z5nlDK8QicZUIbz0,4756
-torch_geometric/hash_tensor.py,sha256=WB-aBCJWNWqnlnzQ8Ob4LHeCXm0u1_NPPhmNAEwBpq4,24906
+torch_geometric/hash_tensor.py,sha256=YtIGQ29nnR1nXp3NGnm4zR7VpqXteqzyVt36IFtZsHc,24922
 torch_geometric/home.py,sha256=EV54B4Dmiv61GDbkCwtCfWGWJ4eFGwZ8s3KOgGjwYgY,790
-torch_geometric/index.py,sha256=ZVt69f_EhDqiFccZgKJ2eGUXMxErQ2u_KbJfXT9lIxI,24328
+torch_geometric/index.py,sha256=FSP-Tkfw6d7P5U8ua8BO2IoEPJSnJT0L2IKyAfIqJR0,24360
 torch_geometric/inspector.py,sha256=nKi5o4Mn6xsG0Ex1GudTEQt_EqnF9mcMqGtp7Shh9sQ,19336
 torch_geometric/isinstance.py,sha256=truZjdU9PxSvjJ6k0d_CLJ2iOpen2o8U-54pbUbNRyE,935
 torch_geometric/lazy_loader.py,sha256=SM0UcXtIdiFge75MKBAWXedoiSOdFDOV0rm1PfoF9cE,908
@@ -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=Ryx6oGoOsEh8rJ3O0O6j8O18ZPHkrIv-7dr-suQZa6Q,15486
+torch_geometric/typing.py,sha256=QspEdob3sBcx9qenTAO7J7KWDTu6ufQ4XYf_eqZ8k8I,15638
 torch_geometric/warnings.py,sha256=SB9dWGovX_KKcxqsOrdTDvSb_j0NoB5vPGnK2vg0jVw,727
 torch_geometric/contrib/__init__.py,sha256=0pWkmXfZtbdr-AKwlii5LTFggTEH-MCrSKpZxrtPlVs,352
 torch_geometric/contrib/datasets/__init__.py,sha256=lrGnWsEiJf5zsBRmshGZZFN_uYR2ezDjbj9n9nCpvtk,23
@@ -99,8 +99,8 @@ torch_geometric/datasets/git_mol_dataset.py,sha256=LsS_dPYUpwhWXMBh17iT7IbjlLOP0
 torch_geometric/datasets/github.py,sha256=Qhqhkvi6eZ8VF_HqP1rL2iYToZavFNsQh7J1WdeM9dA,2687
 torch_geometric/datasets/gnn_benchmark_dataset.py,sha256=4P8n7czF-gf1egLYlAcSSvfB0GXIKpAbH5UjsuFld1M,6976
 torch_geometric/datasets/heterophilous_graph_dataset.py,sha256=yHHtwl4uPrid0vPOxvPV3sIS8HWdswar8FJ0h0OQ9is,4224
-torch_geometric/datasets/hgb_dataset.py,sha256=nKVF05OeMHaGt3IFH3xxcfbTSdlwIY2pUoCHUooMPhQ,8810
-torch_geometric/datasets/hm.py,sha256=cjOkFkBVD_IN29ttDURxnLJohPEtVavkMhFX_GfJX2A,6763
+torch_geometric/datasets/hgb_dataset.py,sha256=TielMHTK2sk8YW8xAFgmlKoJSyArqVhcL51bVs1VTJc,8812
+torch_geometric/datasets/hm.py,sha256=Tpw7JNZmd_d-E5dc0RajMI1vRE3haR1h0xhOxQpkxis,6779
 torch_geometric/datasets/hydro_net.py,sha256=7dEH7Vgfwa-BxkpkXdIx3LvmudJhCR17omkpvPm62dg,11417
 torch_geometric/datasets/icews.py,sha256=Vdlk-PD10AU68xq8X5IOgrK0wgIBFq8A0D6_WtrXiEo,4735
 torch_geometric/datasets/igmc_dataset.py,sha256=pMiOoXjvqhfsDDNw51WT_IVi6wGJ0cUNwTdpEprPh3E,4611
@@ -139,7 +139,7 @@ torch_geometric/datasets/pcqm4m.py,sha256=7ID_xXXIAyuNzYLI2lBWygZl9wGos-dbaz1b6E
 torch_geometric/datasets/planetoid.py,sha256=RksfwR_PI7qGVphs-T-4jXDepYwQCweMXElLm096hgg,7201
 torch_geometric/datasets/polblogs.py,sha256=IYzsvd4R0OojmOOZUoOdCwQYfwcTfth1PNtcBK1yOGc,3045
 torch_geometric/datasets/ppi.py,sha256=zPtg-omC7WYvr9Tzwkb7zNjpXLODsvxKxKdGEUswp2E,5030
-torch_geometric/datasets/qm7.py,sha256=RbDRYPxjYSPukw3lqxMOCQQQRysiW4hRNB_tG2MF0Ag,3325
+torch_geometric/datasets/qm7.py,sha256=bYyK8xlh9kTr5vqueNbLu9EAjIXkQH1KX1VWnjKfOJc,3323
 torch_geometric/datasets/qm9.py,sha256=XU2HTPbgJJ_6hT--X0J2xkXliCbt7_-hub9nuIUQlug,17213
 torch_geometric/datasets/rcdd.py,sha256=gvOoM1tw_X5QMyBB4FkMUwNErMXAvImyjz5twktBAh8,5317
 torch_geometric/datasets/reddit.py,sha256=QUgiKTaj6YTOYbgWgqV8mPYsctOui2ujaM8f8qy81v0,3131
@@ -149,7 +149,7 @@ torch_geometric/datasets/s3dis.py,sha256=_r9vSX8prt4q_N_4xry_Mwqyf1vXIptAiONrt_6
 torch_geometric/datasets/sbm_dataset.py,sha256=-dORNkinyxGwaPtCVpudnktzMppaChTraqwYd6FA6DM,8816
 torch_geometric/datasets/shapenet.py,sha256=tn3HiQQAr6lxHrqxfOVaAtl40guwFYTXWCbSoLfbB8M,8496
 torch_geometric/datasets/shrec2016.py,sha256=cTLhctbqE0EUEvKddJFhPzDb1oLKXOth4O_WzsWtyMk,6323
-torch_geometric/datasets/snap_dataset.py,sha256=r3sC-dHDouyaYoHGdoBY0uO0qOOvD6_Hb96d2ceGMZk,9433
+torch_geometric/datasets/snap_dataset.py,sha256=deJvB6cpIQ3bu_pcWoqgEo1-Kl_NcFi7ZSUci645X0U,9481
 torch_geometric/datasets/suite_sparse.py,sha256=eqjH4vAUq872qdk3YdLkZSwlu6r7HHpTgK0vEVGmY1s,3278
 torch_geometric/datasets/tag_dataset.py,sha256=MbnVCJcryNys1cjdMzXbuk-Rn-BjLbHWFGHx4QYhhUg,14760
 torch_geometric/datasets/taobao.py,sha256=CUcZpbWsNTasevflO8zqP0YvENy89P7wpKS4MHaDJ6Q,4170
@@ -527,7 +527,7 @@ torch_geometric/testing/graph_store.py,sha256=00B7QToCIspYmgN7svQKp1iU-qAzEtrt3V
 torch_geometric/transforms/__init__.py,sha256=P0R2CFg9pXxjTX4NnYfNPrifRPAw5lVXEOxO80q-1Ek,4296
 torch_geometric/transforms/add_gpse.py,sha256=ex7cSLmcSIn-dC5gbic-CLwKegV5Is7Y8mUn-mSOWXg,1555
 torch_geometric/transforms/add_metapaths.py,sha256=GabaPRvUnpFrZJsxLMUBY2Egzx94GTgsMxegL_qTtbk,14239
-torch_geometric/transforms/add_positional_encoding.py,sha256=tuilyubAn3yeyz8mvFc5zxXTlNzh8okKzG9AE2lPG1Q,6049
+torch_geometric/transforms/add_positional_encoding.py,sha256=J5dmdrpZ7Qc2n2rcvsAz084uhMGPTtD1xGl973vdiHY,6033
 torch_geometric/transforms/add_remaining_self_loops.py,sha256=ItU5FAcE-mkbp_wqTLkRhv0RShR5JVr8vr9d5xv3_Ak,2085
 torch_geometric/transforms/add_self_loops.py,sha256=No8-tMqERQdWVHwEOaYr9aeg1A_RLisiidEy-1wzoV8,2024
 torch_geometric/transforms/base_transform.py,sha256=5y4X5JmpKrJsj9XQ8v_CYPcDB83pq7b1g5RLjeBrxWg,1298
@@ -609,7 +609,7 @@ torch_geometric/utils/_to_dense_adj.py,sha256=hl1sboUBvED5Er66bqLms4VdmxKA-7Y3oz
 torch_geometric/utils/_to_dense_batch.py,sha256=-K5NjjfvjKYKJQ3kXgNIDR7lwMJ_GGISI45b50IGMvY,4582
 torch_geometric/utils/_train_test_split_edges.py,sha256=KnBDgnaKuJYTHUOIlvFtzvkHUe-93DG3ckST4-wOERM,3569
 torch_geometric/utils/_tree_decomposition.py,sha256=ZtpjPQJgXbQWtSWjo-Fmhrov0DGO69TfQb9oBFvZ6dM,5304
-torch_geometric/utils/_trim_to_layer.py,sha256=JRZjpIMh6Z4g3k3yo9zdpS8YXMZHFB2q3iK4_Lp1O0c,8307
+torch_geometric/utils/_trim_to_layer.py,sha256=cauOEzMJJK4w9BC-Pg1bHVncBYqG9XxQex3rn10BFjc,8339
 torch_geometric/utils/_unbatch.py,sha256=B0vjKI96PtHvSBG8F_lqvsiJE134aVjUurPZsG6UZRI,2378
 torch_geometric/utils/augmentation.py,sha256=1F0YCuaklZ9ZbXxdFV0oOoemWvLd8p60WvFo2chzl7E,8600
 torch_geometric/utils/convert.py,sha256=0KEJoBOzU-w-mMQu9QYaMhUqcrGBxBmeRl0hv8NPvII,21697
@@ -638,7 +638,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.dev20250530.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
-pyg_nightly-2.7.0.dev20250530.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-pyg_nightly-2.7.0.dev20250530.dist-info/METADATA,sha256=VSryXBgX_QXulsojYmppUE0PozqRG89FJ6dx6e7Qkh0,62952
-pyg_nightly-2.7.0.dev20250530.dist-info/RECORD,,
+pyg_nightly-2.7.0.dev20250531.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
+pyg_nightly-2.7.0.dev20250531.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+pyg_nightly-2.7.0.dev20250531.dist-info/METADATA,sha256=64gHJ8lq6nkXhv-ggwHjW8aPjmjEMCU3n2MrZQeJYFo,62967
+pyg_nightly-2.7.0.dev20250531.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.dev20250530'
+__version__ = '2.7.0.dev20250531'
 
 __all__ = [
     'Index',

torch_geometric/datasets/hgb_dataset.py

@@ -123,8 +123,8 @@ class HGBDataset(InMemoryDataset):
             start = info.index('LINK\tSTART\tEND\tMEANING') + 1
             end = info[start:].index('')
             for key, row in enumerate(info[start:start + end]):
-                row = row.split('\t')[1:]
-                src, dst, rel = (v for v in row if v != '')
+                edge = row.split('\t')[1:]
+                src, dst, rel = (v for v in edge if v != '')
                 src, dst = n_types[int(src)], n_types[int(dst)]
                 rel = rel.split('-')[1]
                 e_types[key] = (src, rel, dst)

torch_geometric/datasets/hm.py

@@ -81,7 +81,7 @@ class HM(InMemoryDataset):
             xs.append(torch.from_numpy(x).to(torch.float))
 
         x = torch.from_numpy(df['age'].values).to(torch.float).view(-1, 1)
-        x = x.nan_to_num(nan=x.nanmean())
+        x = x.nan_to_num(nan=x.nanmean())  # type: ignore
         xs.append(x / x.max())
 
         data['customer'].x = torch.cat(xs, dim=-1)

torch_geometric/datasets/qm7.py

@@ -84,7 +84,7 @@ class QM7b(InMemoryDataset):
             edge_attr = coulomb_matrix[i, edge_index[0], edge_index[1]]
             y = target[i].view(1, -1)
             data = Data(edge_index=edge_index, edge_attr=edge_attr, y=y)
-            data.num_nodes = edge_index.max().item() + 1
+            data.num_nodes = int(edge_index.max()) + 1
             data_list.append(data)
 
         if self.pre_filter is not None:

torch_geometric/datasets/snap_dataset.py

@@ -109,7 +109,7 @@ def read_ego(files: List[str], name: str) -> List[EgoData]:
         row = torch.cat([row, row_ego, col_ego], dim=0)
         col = torch.cat([col, col_ego, row_ego], dim=0)
         edge_index = torch.stack([row, col], dim=0)
-        edge_index = coalesce(edge_index, num_nodes=N)
+        edge_index = coalesce(edge_index, num_nodes=int(N))
 
         data = EgoData(x=x, edge_index=edge_index, circle=circle,
                        circle_batch=circle_batch)
@@ -129,7 +129,7 @@ def read_soc(files: List[str], name: str) -> List[Data]:
     edge_index = pd.read_csv(files[0], sep='\t', header=None,
                              skiprows=skiprows, dtype=np.int64)
     edge_index = torch.from_numpy(edge_index.values).t()
-    num_nodes = edge_index.max().item() + 1
+    num_nodes = int(edge_index.max()) + 1
     edge_index = coalesce(edge_index, num_nodes=num_nodes)
 
     return [Data(edge_index=edge_index, num_nodes=num_nodes)]
@@ -143,11 +143,15 @@ def read_wiki(files: List[str], name: str) -> List[Data]:
     edge_index = torch.from_numpy(edge_index.values).t()
 
     idx = torch.unique(edge_index.flatten())
-    idx_assoc = torch.full((edge_index.max() + 1, ), -1, dtype=torch.long)
+    idx_assoc = torch.full(
+        (edge_index.max() + 1, ),  # type: ignore
+        -1,
+        dtype=torch.long,
+    )
     idx_assoc[idx] = torch.arange(idx.size(0))
 
     edge_index = idx_assoc[edge_index]
-    num_nodes = edge_index.max().item() + 1
+    num_nodes = int(edge_index.max()) + 1
     edge_index = coalesce(edge_index, num_nodes=num_nodes)
 
     return [Data(edge_index=edge_index, num_nodes=num_nodes)]

torch_geometric/edge_index.py

@@ -803,7 +803,7 @@ class EdgeIndex(Tensor):
 
         size = self.get_sparse_size()
         if value is not None and value.dim() > 1:
-            size = size + value.size()[1:]  # type: ignore
+            size = size + value.size()[1:]
 
         out = torch.full(size, fill_value, dtype=dtype, device=self.device)
         out[self._data[0], self._data[1]] = value if value is not None else 1
@@ -1186,10 +1186,10 @@ class EdgeIndex(Tensor):
         return edge_index
 
     # Prevent auto-wrapping outputs back into the proper subclass type:
-    __torch_function__ = torch._C._disabled_torch_function_impl
+    __torch_function__ = torch._C._disabled_torch_function_impl  # type: ignore
 
     @classmethod
-    def __torch_dispatch__(
+    def __torch_dispatch__(  # type: ignore
         cls: Type,
         func: Callable[..., Any],
         types: Iterable[Type[Any]],

torch_geometric/hash_tensor.py

@@ -326,10 +326,10 @@ class HashTensor(Tensor):
     # PyTorch/Python builtins #################################################
 
     # Prevent auto-wrapping outputs back into the proper subclass type:
-    __torch_function__ = torch._C._disabled_torch_function_impl
+    __torch_function__ = torch._C._disabled_torch_function_impl  # type: ignore
 
     @classmethod
-    def __torch_dispatch__(
+    def __torch_dispatch__(  # type: ignore
         cls: Type,
         func: Callable[..., Any],
         types: Iterable[Type[Any]],
@@ -416,7 +416,7 @@ class HashTensor(Tensor):
         """"""  # noqa: D419
         return self._min_key.is_shared()
 
-    def detach_(self) -> 'HashTensor':  # type: ignore
+    def detach_(self) -> 'HashTensor':
         """"""  # noqa: D419
         if self._value is not None:
             self._value.detach_()

torch_geometric/index.py

@@ -361,10 +361,10 @@ class Index(Tensor):
         return index
 
     # Prevent auto-wrapping outputs back into the proper subclass type:
-    __torch_function__ = torch._C._disabled_torch_function_impl
+    __torch_function__ = torch._C._disabled_torch_function_impl  # type: ignore
 
     @classmethod
-    def __torch_dispatch__(
+    def __torch_dispatch__(  # type: ignore
         cls: Type,
         func: Callable[..., Any],
         types: Iterable[Type[Any]],

torch_geometric/transforms/add_positional_encoding.py

@@ -97,7 +97,7 @@ class AddLaplacianEigenvectorPE(BaseTransform):
             from scipy.sparse.linalg import eigs, eigsh
             eig_fn = eigs if not self.is_undirected else eigsh
 
-            eig_vals, eig_vecs = eig_fn(  # type: ignore
+            eig_vals, eig_vecs = eig_fn(
                 L,
                 k=self.k + 1,
                 which='SR' if not self.is_undirected else 'SA',

torch_geometric/typing.py

@@ -10,7 +10,7 @@ import torch
 from torch import Tensor
 
 try:
-    from typing import TypeAlias  # type: ignore
+    from typing import TypeAlias
 except ImportError:
     from typing_extensions import TypeAlias
 
@@ -21,6 +21,8 @@ WITH_PT23 = WITH_PT20 and int(torch.__version__.split('.')[1]) >= 3
 WITH_PT24 = WITH_PT20 and int(torch.__version__.split('.')[1]) >= 4
 WITH_PT25 = WITH_PT20 and int(torch.__version__.split('.')[1]) >= 5
 WITH_PT26 = WITH_PT20 and int(torch.__version__.split('.')[1]) >= 6
+WITH_PT27 = WITH_PT20 and int(torch.__version__.split('.')[1]) >= 7
+WITH_PT28 = WITH_PT20 and int(torch.__version__.split('.')[1]) >= 8
 WITH_PT113 = WITH_PT20 or int(torch.__version__.split('.')[1]) >= 13
 
 WITH_WINDOWS = os.name == 'nt'
@@ -95,7 +97,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
 else:
 
     class CPUHashMap:  # type: ignore
@@ -107,7 +109,7 @@ else:
 
 
 if WITH_CUDA_HASH_MAP:
-    CUDAHashMap: TypeAlias = torch.classes.pyg.CUDAHashMap
+    CUDAHashMap: TypeAlias = torch.classes.pyg.CUDAHashMap  # type: ignore
 else:
 
     class CUDAHashMap:  # type: ignore

torch_geometric/utils/_trim_to_layer.py

@@ -234,10 +234,10 @@ def trim_sparse_tensor(src: SparseTensor, size: Tuple[int, int],
     rowptr = torch.narrow(rowptr, 0, 0, size[0] + 1).clone()
     rowptr[num_seed_nodes + 1:] = rowptr[num_seed_nodes]
 
-    col = torch.narrow(col, 0, 0, rowptr[-1])
+    col = torch.narrow(col, 0, 0, rowptr[-1])  # type: ignore
 
     if value is not None:
-        value = torch.narrow(value, 0, 0, rowptr[-1])
+        value = torch.narrow(value, 0, 0, rowptr[-1])  # type: ignore
 
     csr2csc = src.storage._csr2csc
     if csr2csc is not None: