From 50f2ad213f6ed0d704384220ab4f166916b6df85 Mon Sep 17 00:00:00 2001 From: "Kai Wang (Victor Kai)" <37533040+kaiwang960112@users.noreply.github.com> Date: Tue, 12 Jul 2022 16:35:58 +0800 Subject: [PATCH] [NFC] polish colossalai/engine/ophooks/utils.py code style (#1256) --- 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 a0ad50bfe..fe08405c8 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():