prompt: We are currently solving the following issue within our repository. Here is the issue text:
  --- BEGIN ISSUE ---
  [tensor] fix some unittests
  
  [tensor] fix some unittests
  
  [tensor] fix some unittests
  
  
  --- END ISSUE ---
  
  Below are some code segments, each from a relevant file. One or more of these files may contain bugs.
  --- BEGIN FILES ---
  Path: `colossalai/context/random/__init__.py`
  Content:
  ```
  1 from ._helper import (seed, set_mode, with_seed, add_seed, get_seeds, get_states, get_current_mode, set_seed_states,
  2                       sync_states, moe_set_seed, reset_seeds)
  3 
  4 __all__ = [
  5     'seed', 'set_mode', 'with_seed', 'add_seed', 'get_seeds', 'get_states', 'get_current_mode', 'set_seed_states',
  6     'sync_states', 'moe_set_seed', 'reset_seeds'
  7 ]
  8
  ```
  --- END FILES ---
  
  Please first localize the bug based on the issue statement, and then generate a patch according to the `git diff` format fenced by three backticks.
  
  Here is an example:
  
  ```diff
  diff --git a/examples/server_async.py b/examples/server_async.py
  --- a/examples/server_async.py
  +++ b/examples/server_async.py
  @@ -313,4 +313,4 @@
   
   
   if __name__ == "__main__":
  -    asyncio.run(run_async_server("."), debug=True)
  +    asyncio.run(run_async_server(), debug=True)
  diff --git a/examples/server_sync.py b/examples/server_sync.py
  --- a/examples/server_sync.py
  +++ b/examples/server_sync.py
  @@ -313,5 +313,5 @@
   
   
   if __name__ == "__main__":
  -    server = run_sync_server(".")
  +    server = run_sync_server()
       server.shutdown()
  
  ```
  
golden_diff: diff --git a/colossalai/context/random/__init__.py b/colossalai/context/random/__init__.py
  --- a/colossalai/context/random/__init__.py
  +++ b/colossalai/context/random/__init__.py
  @@ -1,5 +1,16 @@
  -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',
  
verification_info: {"golden_diff": "diff --git a/colossalai/context/random/__init__.py b/colossalai/context/random/__init__.py\n--- a/colossalai/context/random/__init__.py\n+++ b/colossalai/context/random/__init__.py\n@@ -1,5 +1,16 @@\n-from ._helper import (seed, set_mode, with_seed, add_seed, get_seeds, get_states, get_current_mode, set_seed_states,\n-                      sync_states, moe_set_seed, reset_seeds)\n+from ._helper import (\n+    add_seed,\n+    get_current_mode,\n+    get_seeds,\n+    get_states,\n+    moe_set_seed,\n+    reset_seeds,\n+    seed,\n+    set_mode,\n+    set_seed_states,\n+    sync_states,\n+    with_seed,\n+)\n \n __all__ = [\n     'seed', 'set_mode', 'with_seed', 'add_seed', 'get_seeds', 'get_states', 'get_current_mode', 'set_seed_states',\n", "issue": "[tensor] fix some unittests\n\n[tensor] fix some unittests\n\n[tensor] fix some unittests\n\n", "before_files": [{"content": "from ._helper import (seed, set_mode, with_seed, add_seed, get_seeds, get_states, get_current_mode, set_seed_states,\n                      sync_states, moe_set_seed, reset_seeds)\n\n__all__ = [\n    'seed', 'set_mode', 'with_seed', 'add_seed', 'get_seeds', 'get_states', 'get_current_mode', 'set_seed_states',\n    'sync_states', 'moe_set_seed', 'reset_seeds'\n]\n", "path": "colossalai/context/random/__init__.py"}], "after_files": [{"content": "from ._helper import (\n    add_seed,\n    get_current_mode,\n    get_seeds,\n    get_states,\n    moe_set_seed,\n    reset_seeds,\n    seed,\n    set_mode,\n    set_seed_states,\n    sync_states,\n    with_seed,\n)\n\n__all__ = [\n    'seed', 'set_mode', 'with_seed', 'add_seed', 'get_seeds', 'get_states', 'get_current_mode', 'set_seed_states',\n    'sync_states', 'moe_set_seed', 'reset_seeds'\n]\n", "path": "colossalai/context/random/__init__.py"}]}