From 0dc0d57faa5fa23de46f1a7ab046b61fe2832abf Mon Sep 17 00:00:00 2001
From: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
Date: Sun, 6 Oct 2024 08:38:09 +0800
Subject: [PATCH] [DistDGL] back-compatible with dgl.dataloading (#7817)

---
 python/dgl/dataloading/dataloader.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/python/dgl/dataloading/dataloader.py b/python/dgl/dataloading/dataloader.py
index 17c9bb4ffed2..587080b5ad4d 100644
--- a/python/dgl/dataloading/dataloader.py
+++ b/python/dgl/dataloading/dataloader.py
@@ -1502,6 +1502,16 @@ def __new__(cls, *args, **kwargs):
         return NewEdgeCollator(*args, **kwargs)
 
 
+def _remove_kwargs_dist(kwargs):
+    """Deprecated."""
+    if "num_workers" in kwargs:
+        del kwargs["num_workers"]
+    if "pin_memory" in kwargs:
+        del kwargs["pin_memory"]
+        print("Distributed DataLoaders do not support pin_memory.")
+    return kwargs
+
+
 class DistDataLoader:
     """Deprecated. Please use :class:`~dgl.distributed.DistDataLoader` instead."""