[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

colossalai/legacy/utils/activation_checkpoint.py

@@ -226,8 +226,10 @@ def inner_unpack(packed):
 
             # rerun forward, the inner_pack will store all the activations in storage
             if has_autocast_in_fwd:
-                with torch.enable_grad(), get_accelerator().autocast()(), torch.autograd.graph.saved_tensors_hooks(
-                    inner_pack, inner_unpack
+                with (
+                    torch.enable_grad(),
+                    get_accelerator().autocast()(),
+                    torch.autograd.graph.saved_tensors_hooks(inner_pack, inner_unpack),
                 ):
                     _unused = function(*args)
             else: