From ec744a2210cdf8d98480d202d2734e111f0f6de7 Mon Sep 17 00:00:00 2001 From: kaiwang960112 Date: Tue, 12 Jul 2022 16:31:38 +0800 Subject: [PATCH] [NFC] polish colossalai/engine/ophooks/utils.py code style --- colossalai/engine/ophooks/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/colossalai/engine/ophooks/utils.py b/colossalai/engine/ophooks/utils.py index a0ad50bfe2eb..fe08405c82bf 100644 --- 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():