flake8 style change (#363)

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)

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(