From 8539898ec62e0c48720370411e5d516ddb85b244 Mon Sep 17 00:00:00 2001
From: RichardoLuo <50363844+RichardoLuo@users.noreply.github.com>
Date: Thu, 10 Mar 2022 11:47:51 +0800
Subject: [PATCH] flake8 style change (#363)

---
 colossalai/context/random/_helper.py      | 2 +-
 colossalai/context/random/seed_manager.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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(