[NFC] polish colossalai/engine/ophooks/utils.py code style (#1256)

2022-07-12 16:35:58 +08:00 · 2022-07-12 16:35:58 +08:00 · 50f2ad213f
parent 2dd4d556fb
commit 50f2ad213f
1 changed files with 1 additions and 1 deletions
--- a/colossalai/engine/ophooks/utils.py
+++ b/colossalai/engine/ophooks/utils.py
@ -93,7 +93,7 @@ def register_ophooks_recursively(module: torch.nn.Module,
    assert isinstance(ophook_list, (list, tuple))
    assert len(ophook_list) > 0, 'expected at least 1 hook in the argument ophook_list but found 0'
    for hook in ophook_list:
-            assert (isinstance(hook, BaseOpHook))
+        assert (isinstance(hook, BaseOpHook))

    # Add hooks for submodules
    for child_name, child in module.named_children():