diff --git a/colossalai/context/random/_helper.py b/colossalai/context/random/_helper.py
index ba5308cdc..456731192 100644
--- a/colossalai/context/random/_helper.py
+++ b/colossalai/context/random/_helper.py
@@ -56,7 +56,7 @@ def add_seed(parallel_mode: ParallelMode, seed: int, overwrite: bool = False):
     :type parallel_mode: :class:`colossalai.context.ParallelMode`
     :param seed: The seed to be added
     :type seed: int
-    :raises AssertionError: Raises an AssertionError if `parallel_mode` is not an instance of 
+    :raises AssertionError: Raises an AssertionError if `parallel_mode` is not an instance of
         :class:`colossalai.context.ParallelMode` or the seed for `parallel_mode` has been added
     """
     _SEED_MANAGER.add_seed(parallel_mode, seed, overwrite)
diff --git a/colossalai/context/random/seed_manager.py b/colossalai/context/random/seed_manager.py
index 02b8a88a6..fae1ce6f2 100644
--- a/colossalai/context/random/seed_manager.py
+++ b/colossalai/context/random/seed_manager.py
@@ -63,7 +63,7 @@ class SeedManager:
         :type seed: int
         :param overwrtie: Whether allows to overwrite the seed that has been set already
         :type overwrtie: bool, optional
-        :raises AssertionError: Raises an AssertionError if `parallel_mode` is not an instance of 
+        :raises AssertionError: Raises an AssertionError if `parallel_mode` is not an instance of
             :class:`colossalai.context.ParallelMode` or the seed for `parallel_mode` has been added
         """
         assert isinstance(