github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-05 18:14:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

flake8 style change (#363)

Browse Source
This commit is contained in:
RichardoLuo 2022-03-10 11:47:51 +08:00 committed by Frank Lee
parent 53bb3bcc0a
commit 8539898ec6
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/context/random/_helper.py
View File

@ -56,7 +56,7 @@ def add_seed(parallel_mode: ParallelMode, seed: int, overwrite: bool = False):
:type parallel_mode: :class:`colossalai.context.ParallelMode` :type parallel_mode: :class:`colossalai.context.ParallelMode`
:param seed: The seed to be added :param seed: The seed to be added
:type seed: int :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 :class:`colossalai.context.ParallelMode` or the seed for `parallel_mode` has been added
""" """
_SEED_MANAGER.add_seed(parallel_mode, seed, overwrite) _SEED_MANAGER.add_seed(parallel_mode, seed, overwrite)

2
colossalai/context/random/seed_manager.py
View File

@ -63,7 +63,7 @@ class SeedManager:
:type seed: int :type seed: int
:param overwrtie: Whether allows to overwrite the seed that has been set already :param overwrtie: Whether allows to overwrite the seed that has been set already
:type overwrtie: bool, optional :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 :class:`colossalai.context.ParallelMode` or the seed for `parallel_mode` has been added
""" """
assert isinstance( assert isinstance(