diff --git a/test/dataloader2/test_dataloader2.py b/test/dataloader2/test_dataloader2.py
index e87e7beff..270fe0eab 100644
--- a/test/dataloader2/test_dataloader2.py
+++ b/test/dataloader2/test_dataloader2.py
@@ -20,7 +20,7 @@
 import torch.distributed as dist
 from torch.testing._internal.common_utils import instantiate_parametrized_tests, parametrize
 
-from torch.utils.data.datapipes.iter.grouping import SHARDING_PRIORITIES
+from torch.utils.data.datapipes.iter.sharding import SHARDING_PRIORITIES
 
 from torchdata.dataloader2 import (
     communication,
diff --git a/test/test_graph.py b/test/test_graph.py
index a06a84d4a..79d1064cb 100644
--- a/test/test_graph.py
+++ b/test/test_graph.py
@@ -14,7 +14,7 @@
 from _utils._common_utils_for_test import IS_WINDOWS
 
 from torch.utils.data import IterDataPipe
-from torch.utils.data.datapipes.iter.grouping import SHARDING_PRIORITIES
+from torch.utils.data.datapipes.iter.sharding import SHARDING_PRIORITIES
 
 from torchdata.dataloader2 import DataLoader2, MultiProcessingReadingService, ReadingServiceInterface
 from torchdata.dataloader2.graph import find_dps, list_dps, remove_dp, replace_dp, traverse_dps
diff --git a/torchdata/dataloader2/reading_service.py b/torchdata/dataloader2/reading_service.py
index 1aeb3582b..439eebf53 100644
--- a/torchdata/dataloader2/reading_service.py
+++ b/torchdata/dataloader2/reading_service.py
@@ -18,7 +18,7 @@
 import torch.multiprocessing as mp
 
 from torch.utils.data import DataLoader
-from torch.utils.data.datapipes.iter.grouping import SHARDING_PRIORITIES
+from torch.utils.data.datapipes.iter.sharding import SHARDING_PRIORITIES
 
 from torchdata._constants import default_dl2_worker_join_timeout_in_s, default_timeout_in_s
 from torchdata.dataloader2 import communication
diff --git a/torchdata/dataloader2/utils/worker.py b/torchdata/dataloader2/utils/worker.py
index 316770d55..6838eddf0 100644
--- a/torchdata/dataloader2/utils/worker.py
+++ b/torchdata/dataloader2/utils/worker.py
@@ -13,7 +13,7 @@
 
 import torch
 
-from torch.utils.data.datapipes.iter.grouping import SHARDING_PRIORITIES
+from torch.utils.data.datapipes.iter.sharding import SHARDING_PRIORITIES
 
 from torchdata.dataloader2 import communication
 from torchdata.dataloader2.graph import (
diff --git a/torchdata/datapipes/iter/__init__.pyi.in b/torchdata/datapipes/iter/__init__.pyi.in
index 855185324..141306b4f 100644
--- a/torchdata/datapipes/iter/__init__.pyi.in
+++ b/torchdata/datapipes/iter/__init__.pyi.in
@@ -12,7 +12,7 @@ from torchdata._constants import default_timeout_in_s
 from torchdata.datapipes.map import MapDataPipe
 from torch.utils.data import DataChunk, IterableDataset, default_collate
 from torch.utils.data.datapipes._typing import _DataPipeMeta
-from torch.utils.data.datapipes.iter.grouping import SHARDING_PRIORITIES
+from torch.utils.data.datapipes.iter.sharding import SHARDING_PRIORITIES
 
 from typing import Any, Callable, Dict, List, Optional, Sequence, TypeVar, Union, Hashable
 
diff --git a/torchdata/datapipes/iter/util/sharding.py b/torchdata/datapipes/iter/util/sharding.py
index 9e043f3aa..9b755b99a 100644
--- a/torchdata/datapipes/iter/util/sharding.py
+++ b/torchdata/datapipes/iter/util/sharding.py
@@ -6,7 +6,7 @@
 
 from typing import Iterator, Optional, TypeVar
 
-from torch.utils.data.datapipes.iter.grouping import SHARDING_PRIORITIES
+from torch.utils.data.datapipes.iter.sharding import SHARDING_PRIORITIES
 from torchdata.datapipes import functional_datapipe
 from torchdata.datapipes.iter import IterDataPipe