[NFC] polish colossalai/context/random/__init__.py code style (#3327)

[NFC] polish colossalai/context/random/init.py code style (#3327)
198a74b9 · yuxuan-lou · GitHub · 82132f4e · 198a74b9
Unverified Commit 198a74b9 authored Mar 30, 2023 by yuxuan-lou Committed by GitHub Mar 30, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 13 additions and 2 deletions

colossalai/context/random/__init__.py colossalai/context/random/__init__.py +13 -2

No files found.
--- a/colossalai/context/random/__init__.py
+++ b/colossalai/context/random/__init__.py
-from ._helper import (seed, set_mode, with_seed, add_seed, get_seeds, get_states, get_current_mode, set_seed_states,
-                      sync_states, moe_set_seed, reset_seeds)
+from ._helper import (
+    add_seed,
+    get_current_mode,
+    get_seeds,
+    get_states,
+    moe_set_seed,
+    reset_seeds,
+    seed,
+    set_mode,
+    set_seed_states,
+    sync_states,
+    with_seed,
+)

 __all__ = [
    'seed', 'set_mode', 'with_seed', 'add_seed', 'get_seeds', 'get_states', 'get_current_mode', 'set_seed_states',