diff --git a/mmengine/data/sampler.py b/mmengine/data/sampler.py
index 83936a2772ff8d89b1243bcc884eb74526c0928f..47b2c3b4c1e3e5ae5a5f1941263f598c3a6af2f2 100644
--- a/mmengine/data/sampler.py
+++ b/mmengine/data/sampler.py
@@ -16,7 +16,7 @@ sync_random_seed = MagicMock(return_value=0)
 
 
 @DATA_SAMPLERS.register_module()
-class DefaultSampler(Sampler[int]):
+class DefaultSampler(Sampler):
     """The default data sampler for both distributed and non-distributed
     environment.
 
@@ -109,7 +109,7 @@ class DefaultSampler(Sampler[int]):
 
 
 @DATA_SAMPLERS.register_module()
-class InfiniteSampler(Sampler[int]):
+class InfiniteSampler(Sampler):
     """It's designed for iteration-based runner and yields a mini-batch indices
     each time.