From abd480daac7277122ac50e8f5b5f2021312aa352 Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Mon, 27 Oct 2025 13:06:39 -0700
Subject: [PATCH 01/21] Update feature_store.py

---
 torch_geometric/llm/utils/feature_store.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch_geometric/llm/utils/feature_store.py b/torch_geometric/llm/utils/feature_store.py
index 005a78f09cd5..4f4b97d01b69 100644
--- a/torch_geometric/llm/utils/feature_store.py
+++ b/torch_geometric/llm/utils/feature_store.py
@@ -6,7 +6,7 @@
 from torch import Tensor
 
 from torch_geometric.data import Data, HeteroData
-from torch_geometric.distributed import LocalFeatureStore
+from torch_geometric.distributed.local_feature_store import LocalFeatureStore
 from torch_geometric.llm.utils.backend_utils import batch_knn
 from torch_geometric.sampler import HeteroSamplerOutput, SamplerOutput
 from torch_geometric.typing import InputNodes

From 91070df9b83ceb73663880da8f3852d19715c98b Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Mon, 27 Oct 2025 13:10:32 -0700
Subject: [PATCH 02/21] Update graph_store.py

---
 torch_geometric/llm/utils/graph_store.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch_geometric/llm/utils/graph_store.py b/torch_geometric/llm/utils/graph_store.py
index fa3fc68391c4..932d5cfb359b 100644
--- a/torch_geometric/llm/utils/graph_store.py
+++ b/torch_geometric/llm/utils/graph_store.py
@@ -4,7 +4,7 @@
 from torch import Tensor
 
 from torch_geometric.data import FeatureStore
-from torch_geometric.distributed import LocalGraphStore
+from torch_geometric.distributed.local_graph_store import LocalGraphStore
 from torch_geometric.sampler import (
     BidirectionalNeighborSampler,
     NodeSamplerInput,

From f90e5639d8842c1c50ea46c13fe0746eee930343 Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 14:47:39 -0700
Subject: [PATCH 03/21] Update __init__.py

---
 torch_geometric/distributed/__init__.py | 11 -----------
 1 file changed, 11 deletions(-)

diff --git a/torch_geometric/distributed/__init__.py b/torch_geometric/distributed/__init__.py
index 767018d95b30..749f6d11762b 100644
--- a/torch_geometric/distributed/__init__.py
+++ b/torch_geometric/distributed/__init__.py
@@ -9,17 +9,6 @@
 from .dist_neighbor_loader import DistNeighborLoader
 from .dist_link_neighbor_loader import DistLinkNeighborLoader
 
-warn(
-    "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
-    "no longer be maintained. For distributed training, refer to our "
-    "tutorials on distributed training at "
-    "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
-    "or cuGraph examples at "
-    "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
-    stacklevel=2,
-    category=DeprecationWarning,
-)
-
 __all__ = classes = [
     'DistContext',
     'LocalFeatureStore',

From 22046645c5515b82a90c1a3658becb6a5e7894d0 Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 14:48:43 -0700
Subject: [PATCH 04/21] Update dist_neighbor_loader.py

---
 torch_geometric/distributed/dist_neighbor_loader.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/torch_geometric/distributed/dist_neighbor_loader.py b/torch_geometric/distributed/dist_neighbor_loader.py
index a99a192405f5..dd0b45af1920 100644
--- a/torch_geometric/distributed/dist_neighbor_loader.py
+++ b/torch_geometric/distributed/dist_neighbor_loader.py
@@ -1,6 +1,7 @@
 from typing import Callable, Dict, List, Optional, Tuple, Union
 
 import torch
+from warnings import warn
 
 from torch_geometric.distributed import (
     DistContext,
@@ -61,6 +62,16 @@ def __init__(
         device: Optional[torch.device] = None,
         **kwargs,
     ):
+        warn(
+            "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
+            "no longer be maintained. For distributed training, refer to our "
+            "tutorials on distributed training at "
+            "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
+            "or cuGraph examples at "
+            "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
+            stacklevel=2,
+            category=DeprecationWarning,
+        )
         assert isinstance(data[0], LocalFeatureStore)
         assert isinstance(data[1], LocalGraphStore)
         assert concurrency >= 1, "RPC concurrency must be greater than 1"

From 63b223762cb9f27c701c06626ccfdf4416bfc37a Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 14:49:27 -0700
Subject: [PATCH 05/21] Update dist_loader.py

---
 torch_geometric/distributed/dist_loader.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/torch_geometric/distributed/dist_loader.py b/torch_geometric/distributed/dist_loader.py
index a67c3aebf1b5..78e79bec81fa 100644
--- a/torch_geometric/distributed/dist_loader.py
+++ b/torch_geometric/distributed/dist_loader.py
@@ -53,6 +53,14 @@ def __init__(
         dist_sampler: DistNeighborSampler = None,
         **kwargs,
     ):
+        warn(
+            "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
+            "no longer be maintained. For distributed training, refer to our "
+            "tutorials on distributed training at "
+            "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
+            "or cuGraph examples at "
+            "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
+        )
         if master_addr is None and os.environ.get('MASTER_ADDR') is not None:
             master_addr = os.environ['MASTER_ADDR']
         if master_addr is None:

From 0c49bdcb3c611de71a497e23dd1cba1b6ecf811e Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 14:49:46 -0700
Subject: [PATCH 06/21] Update dist_loader.py

---
 torch_geometric/distributed/dist_loader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch_geometric/distributed/dist_loader.py b/torch_geometric/distributed/dist_loader.py
index 78e79bec81fa..a3545f8bd452 100644
--- a/torch_geometric/distributed/dist_loader.py
+++ b/torch_geometric/distributed/dist_loader.py
@@ -2,7 +2,7 @@
 import logging
 import os
 from typing import Any, Optional, Union
-
+from warnings import warn
 import torch.distributed
 import torch.multiprocessing as mp
 

From b853ab7b694bdf5a30969adcf8ba8970d7dcc628 Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 14:50:04 -0700
Subject: [PATCH 07/21] Update dist_neighbor_loader.py

---
 torch_geometric/distributed/dist_neighbor_loader.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/torch_geometric/distributed/dist_neighbor_loader.py b/torch_geometric/distributed/dist_neighbor_loader.py
index dd0b45af1920..060c5cba614d 100644
--- a/torch_geometric/distributed/dist_neighbor_loader.py
+++ b/torch_geometric/distributed/dist_neighbor_loader.py
@@ -69,8 +69,6 @@ def __init__(
             "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
             "or cuGraph examples at "
             "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
-            stacklevel=2,
-            category=DeprecationWarning,
         )
         assert isinstance(data[0], LocalFeatureStore)
         assert isinstance(data[1], LocalGraphStore)

From 6a4709a60e0bfb4b0c9d05c3d54c0953f4149b43 Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 14:50:25 -0700
Subject: [PATCH 08/21] Update dist_neighbor_sampler.py

---
 torch_geometric/distributed/dist_neighbor_sampler.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/torch_geometric/distributed/dist_neighbor_sampler.py b/torch_geometric/distributed/dist_neighbor_sampler.py
index e8433fa7b57b..5641d6935440 100644
--- a/torch_geometric/distributed/dist_neighbor_sampler.py
+++ b/torch_geometric/distributed/dist_neighbor_sampler.py
@@ -2,7 +2,7 @@
 import logging
 import math
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
-
+from warnings import warn
 import numpy as np
 import torch
 import torch.multiprocessing as mp
@@ -81,6 +81,14 @@ def __init__(
         device: Optional[torch.device] = None,
         **kwargs,
     ):
+        warn(
+            "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
+            "no longer be maintained. For distributed training, refer to our "
+            "tutorials on distributed training at "
+            "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
+            "or cuGraph examples at "
+            "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
+        )
         self.current_ctx = current_ctx
 
         self.feature_store, self.graph_store = data

From 6ab12d20eacfddb639e96e021c6dbf588192b57f Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Tue, 28 Oct 2025 21:50:44 +0000
Subject: [PATCH 09/21] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 torch_geometric/distributed/dist_neighbor_loader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch_geometric/distributed/dist_neighbor_loader.py b/torch_geometric/distributed/dist_neighbor_loader.py
index 060c5cba614d..79a762ba4201 100644
--- a/torch_geometric/distributed/dist_neighbor_loader.py
+++ b/torch_geometric/distributed/dist_neighbor_loader.py
@@ -1,7 +1,7 @@
 from typing import Callable, Dict, List, Optional, Tuple, Union
+from warnings import warn
 
 import torch
-from warnings import warn
 
 from torch_geometric.distributed import (
     DistContext,

From 00e4712341f97129a056e10d3ecaa79e3d3d5a58 Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 14:51:19 -0700
Subject: [PATCH 10/21] Update dist_neighbor_loader.py

---
 .../distributed/dist_neighbor_loader.py         | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/torch_geometric/distributed/dist_neighbor_loader.py b/torch_geometric/distributed/dist_neighbor_loader.py
index 79a762ba4201..83d1069e76e1 100644
--- a/torch_geometric/distributed/dist_neighbor_loader.py
+++ b/torch_geometric/distributed/dist_neighbor_loader.py
@@ -62,14 +62,15 @@ def __init__(
         device: Optional[torch.device] = None,
         **kwargs,
     ):
-        warn(
-            "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
-            "no longer be maintained. For distributed training, refer to our "
-            "tutorials on distributed training at "
-            "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
-            "or cuGraph examples at "
-            "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
-        )
+        if dist_sampler is not None:
+            warn(
+                "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
+                "no longer be maintained. For distributed training, refer to our "
+                "tutorials on distributed training at "
+                "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
+                "or cuGraph examples at "
+                "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
+            )
         assert isinstance(data[0], LocalFeatureStore)
         assert isinstance(data[1], LocalGraphStore)
         assert concurrency >= 1, "RPC concurrency must be greater than 1"

From c3214f908b05776d1644e96529cb7d860e5a4d55 Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 14:52:50 -0700
Subject: [PATCH 11/21] Update dist_link_neighbor_loader.py

---
 torch_geometric/distributed/dist_link_neighbor_loader.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/torch_geometric/distributed/dist_link_neighbor_loader.py b/torch_geometric/distributed/dist_link_neighbor_loader.py
index e62deb1e6dee..f8a85c71c585 100644
--- a/torch_geometric/distributed/dist_link_neighbor_loader.py
+++ b/torch_geometric/distributed/dist_link_neighbor_loader.py
@@ -93,6 +93,15 @@ def __init__(
                 channel=channel,
                 concurrency=concurrency,
             )
+        else:
+            warn(
+                "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
+                "no longer be maintained. For distributed training, refer to our "
+                "tutorials on distributed training at "
+                "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
+                "or cuGraph examples at "
+                "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
+            )
 
         DistLoader.__init__(
             self,

From 66ab13b4376d1afc710f9af4de768f4261455859 Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 14:53:12 -0700
Subject: [PATCH 12/21] Update __init__.py

---
 torch_geometric/distributed/__init__.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/torch_geometric/distributed/__init__.py b/torch_geometric/distributed/__init__.py
index 749f6d11762b..d452b2a7fce3 100644
--- a/torch_geometric/distributed/__init__.py
+++ b/torch_geometric/distributed/__init__.py
@@ -1,5 +1,3 @@
-from warnings import warn
-
 from .dist_context import DistContext
 from .local_feature_store import LocalFeatureStore
 from .local_graph_store import LocalGraphStore

From ba3eabe624668f28d203821269804413d212100a Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 14:54:27 -0700
Subject: [PATCH 13/21] Update dist_neighbor_loader.py

---
 .../distributed/dist_neighbor_loader.py        | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/torch_geometric/distributed/dist_neighbor_loader.py b/torch_geometric/distributed/dist_neighbor_loader.py
index 83d1069e76e1..7f55a5c617d6 100644
--- a/torch_geometric/distributed/dist_neighbor_loader.py
+++ b/torch_geometric/distributed/dist_neighbor_loader.py
@@ -62,15 +62,6 @@ def __init__(
         device: Optional[torch.device] = None,
         **kwargs,
     ):
-        if dist_sampler is not None:
-            warn(
-                "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
-                "no longer be maintained. For distributed training, refer to our "
-                "tutorials on distributed training at "
-                "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
-                "or cuGraph examples at "
-                "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
-            )
         assert isinstance(data[0], LocalFeatureStore)
         assert isinstance(data[1], LocalGraphStore)
         assert concurrency >= 1, "RPC concurrency must be greater than 1"
@@ -96,6 +87,15 @@ def __init__(
                 channel=channel,
                 concurrency=concurrency,
             )
+        else:
+            warn(
+                "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
+                "no longer be maintained. For distributed training, refer to our "
+                "tutorials on distributed training at "
+                "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
+                "or cuGraph examples at "
+                "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
+            )
 
         DistLoader.__init__(
             self,

From 401e47410dad9e23eda8c052b863ed097257e7c4 Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 14:54:43 -0700
Subject: [PATCH 14/21] Update dist_loader.py

---
 torch_geometric/distributed/dist_loader.py | 8 --------
 1 file changed, 8 deletions(-)

diff --git a/torch_geometric/distributed/dist_loader.py b/torch_geometric/distributed/dist_loader.py
index a3545f8bd452..a3f627a4e691 100644
--- a/torch_geometric/distributed/dist_loader.py
+++ b/torch_geometric/distributed/dist_loader.py
@@ -53,14 +53,6 @@ def __init__(
         dist_sampler: DistNeighborSampler = None,
         **kwargs,
     ):
-        warn(
-            "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
-            "no longer be maintained. For distributed training, refer to our "
-            "tutorials on distributed training at "
-            "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
-            "or cuGraph examples at "
-            "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
-        )
         if master_addr is None and os.environ.get('MASTER_ADDR') is not None:
             master_addr = os.environ['MASTER_ADDR']
         if master_addr is None:

From b7da2c06da4a969b7a04fc4895ebc008890c44ef Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Tue, 28 Oct 2025 21:58:57 +0000
Subject: [PATCH 15/21] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 torch_geometric/distributed/dist_loader.py           | 2 +-
 torch_geometric/distributed/dist_neighbor_sampler.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/torch_geometric/distributed/dist_loader.py b/torch_geometric/distributed/dist_loader.py
index a3f627a4e691..a67c3aebf1b5 100644
--- a/torch_geometric/distributed/dist_loader.py
+++ b/torch_geometric/distributed/dist_loader.py
@@ -2,7 +2,7 @@
 import logging
 import os
 from typing import Any, Optional, Union
-from warnings import warn
+
 import torch.distributed
 import torch.multiprocessing as mp
 
diff --git a/torch_geometric/distributed/dist_neighbor_sampler.py b/torch_geometric/distributed/dist_neighbor_sampler.py
index 5641d6935440..7dd2625ea3ad 100644
--- a/torch_geometric/distributed/dist_neighbor_sampler.py
+++ b/torch_geometric/distributed/dist_neighbor_sampler.py
@@ -3,6 +3,7 @@
 import math
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 from warnings import warn
+
 import numpy as np
 import torch
 import torch.multiprocessing as mp

From d08f1835a6e4f59d922c7963603a49a70897864e Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 18:42:17 -0700
Subject: [PATCH 16/21] Update dist_link_neighbor_loader.py

---
 torch_geometric/distributed/dist_link_neighbor_loader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch_geometric/distributed/dist_link_neighbor_loader.py b/torch_geometric/distributed/dist_link_neighbor_loader.py
index f8a85c71c585..d18cc90d6702 100644
--- a/torch_geometric/distributed/dist_link_neighbor_loader.py
+++ b/torch_geometric/distributed/dist_link_neighbor_loader.py
@@ -1,5 +1,5 @@
 from typing import Callable, Dict, List, Optional, Tuple, Union
-
+from warnings import warn
 import torch
 
 from torch_geometric.distributed import (

From 4d363da4c7e38a69df9547f4e3398978363e68ee Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 18:44:07 -0700
Subject: [PATCH 17/21] Update dist_link_neighbor_loader.py

---
 torch_geometric/distributed/dist_link_neighbor_loader.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/torch_geometric/distributed/dist_link_neighbor_loader.py b/torch_geometric/distributed/dist_link_neighbor_loader.py
index d18cc90d6702..17e05b4bab43 100644
--- a/torch_geometric/distributed/dist_link_neighbor_loader.py
+++ b/torch_geometric/distributed/dist_link_neighbor_loader.py
@@ -95,8 +95,8 @@ def __init__(
             )
         else:
             warn(
-                "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
-                "no longer be maintained. For distributed training, refer to our "
+                "`torch_geometric.distributed` has been deprecated since 2.7.0 and will " # noqa: E501
+                "no longer be maintained. For distributed training, refer to our " # noqa: E501
                 "tutorials on distributed training at "
                 "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
                 "or cuGraph examples at "

From bf195c9680bea05e6b467abb5cbcc1cb1297c7ba Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 18:44:26 -0700
Subject: [PATCH 18/21] Update dist_neighbor_loader.py

---
 torch_geometric/distributed/dist_neighbor_loader.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/torch_geometric/distributed/dist_neighbor_loader.py b/torch_geometric/distributed/dist_neighbor_loader.py
index 7f55a5c617d6..9ab3875e3586 100644
--- a/torch_geometric/distributed/dist_neighbor_loader.py
+++ b/torch_geometric/distributed/dist_neighbor_loader.py
@@ -89,8 +89,8 @@ def __init__(
             )
         else:
             warn(
-                "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
-                "no longer be maintained. For distributed training, refer to our "
+                "`torch_geometric.distributed` has been deprecated since 2.7.0 and will " # noqa: E501
+                "no longer be maintained. For distributed training, refer to our " # noqa: E501
                 "tutorials on distributed training at "
                 "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
                 "or cuGraph examples at "

From c6b6c0363fb2b37481390bb8db478b8864ed4afc Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 29 Oct 2025 01:44:39 +0000
Subject: [PATCH 19/21] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 torch_geometric/distributed/dist_link_neighbor_loader.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/torch_geometric/distributed/dist_link_neighbor_loader.py b/torch_geometric/distributed/dist_link_neighbor_loader.py
index 17e05b4bab43..54b01e36da28 100644
--- a/torch_geometric/distributed/dist_link_neighbor_loader.py
+++ b/torch_geometric/distributed/dist_link_neighbor_loader.py
@@ -1,5 +1,6 @@
 from typing import Callable, Dict, List, Optional, Tuple, Union
 from warnings import warn
+
 import torch
 
 from torch_geometric.distributed import (

From aa8038803e7c9bb082f27fe1ddeaace8b476e998 Mon Sep 17 00:00:00 2001
From: Rishi Puri <riship@nvidia.com>
Date: Tue, 28 Oct 2025 18:45:18 -0700
Subject: [PATCH 20/21] Update dist_neighbor_sampler.py

---
 torch_geometric/distributed/dist_neighbor_sampler.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/torch_geometric/distributed/dist_neighbor_sampler.py b/torch_geometric/distributed/dist_neighbor_sampler.py
index 7dd2625ea3ad..eed6fe6db2e8 100644
--- a/torch_geometric/distributed/dist_neighbor_sampler.py
+++ b/torch_geometric/distributed/dist_neighbor_sampler.py
@@ -83,12 +83,12 @@ def __init__(
         **kwargs,
     ):
         warn(
-            "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "
-            "no longer be maintained. For distributed training, refer to our "
+            "`torch_geometric.distributed` has been deprecated since 2.7.0 and will " # noqa: E501
+            "no longer be maintained. For distributed training, refer to our " # noqa: E501
             "tutorials on distributed training at "
-            "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
+            "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html " # noqa: E501
             "or cuGraph examples at "
-            "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
+            "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples", # noqa: E501
         )
         self.current_ctx = current_ctx
 

From 0546ccdd7d5760a74ab8f1d586f528efafff526e Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 29 Oct 2025 01:47:44 +0000
Subject: [PATCH 21/21] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 torch_geometric/distributed/dist_link_neighbor_loader.py | 4 ++--
 torch_geometric/distributed/dist_neighbor_loader.py      | 4 ++--
 torch_geometric/distributed/dist_neighbor_sampler.py     | 8 ++++----
 3 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/torch_geometric/distributed/dist_link_neighbor_loader.py b/torch_geometric/distributed/dist_link_neighbor_loader.py
index 54b01e36da28..8a371c21bfdf 100644
--- a/torch_geometric/distributed/dist_link_neighbor_loader.py
+++ b/torch_geometric/distributed/dist_link_neighbor_loader.py
@@ -96,8 +96,8 @@ def __init__(
             )
         else:
             warn(
-                "`torch_geometric.distributed` has been deprecated since 2.7.0 and will " # noqa: E501
-                "no longer be maintained. For distributed training, refer to our " # noqa: E501
+                "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "  # noqa: E501
+                "no longer be maintained. For distributed training, refer to our "  # noqa: E501
                 "tutorials on distributed training at "
                 "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
                 "or cuGraph examples at "
diff --git a/torch_geometric/distributed/dist_neighbor_loader.py b/torch_geometric/distributed/dist_neighbor_loader.py
index 9ab3875e3586..173fbde430bc 100644
--- a/torch_geometric/distributed/dist_neighbor_loader.py
+++ b/torch_geometric/distributed/dist_neighbor_loader.py
@@ -89,8 +89,8 @@ def __init__(
             )
         else:
             warn(
-                "`torch_geometric.distributed` has been deprecated since 2.7.0 and will " # noqa: E501
-                "no longer be maintained. For distributed training, refer to our " # noqa: E501
+                "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "  # noqa: E501
+                "no longer be maintained. For distributed training, refer to our "  # noqa: E501
                 "tutorials on distributed training at "
                 "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
                 "or cuGraph examples at "
diff --git a/torch_geometric/distributed/dist_neighbor_sampler.py b/torch_geometric/distributed/dist_neighbor_sampler.py
index eed6fe6db2e8..28567bf89285 100644
--- a/torch_geometric/distributed/dist_neighbor_sampler.py
+++ b/torch_geometric/distributed/dist_neighbor_sampler.py
@@ -83,12 +83,12 @@ def __init__(
         **kwargs,
     ):
         warn(
-            "`torch_geometric.distributed` has been deprecated since 2.7.0 and will " # noqa: E501
-            "no longer be maintained. For distributed training, refer to our " # noqa: E501
+            "`torch_geometric.distributed` has been deprecated since 2.7.0 and will "  # noqa: E501
+            "no longer be maintained. For distributed training, refer to our "  # noqa: E501
             "tutorials on distributed training at "
-            "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html " # noqa: E501
+            "https://pytorch-geometric.readthedocs.io/en/latest/tutorial/distributed.html "  # noqa: E501
             "or cuGraph examples at "
-            "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples", # noqa: E501
+            "https://github.com/rapidsai/cugraph-gnn/tree/main/python/cugraph-pyg/cugraph_pyg/examples",  # noqa: E501
         )
         self.current_ctx = current_ctx