From 7aec0648ddba9df9dcf8b57a5cd35fb23bcc8ddb Mon Sep 17 00:00:00 2001 From: Vincent Wang Date: Wed, 26 Oct 2022 12:24:33 +0800 Subject: [PATCH] fix policheck --- .../hierarchical_ortmodule/_hierarchical_ortmodule.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/orttraining/orttraining/python/training/ortmodule/experimental/hierarchical_ortmodule/_hierarchical_ortmodule.py b/orttraining/orttraining/python/training/ortmodule/experimental/hierarchical_ortmodule/_hierarchical_ortmodule.py index 06f024111dc6c..adb2edefad4c7 100644 --- a/orttraining/orttraining/python/training/ortmodule/experimental/hierarchical_ortmodule/_hierarchical_ortmodule.py +++ b/orttraining/orttraining/python/training/ortmodule/experimental/hierarchical_ortmodule/_hierarchical_ortmodule.py @@ -112,7 +112,7 @@ def record_args(module, args): # those nn.Module's with their recorded inputs. # NOTE that if a module is not called from forward(), it will fail to be captured by this hook. def recursive_hook(module): - # We cannot skip module in whitelist because it's possible that a module is called multiple times + # We cannot skip module in allowlist because it's possible that a module is called multiple times # so that we still need to know the number of different input sets and use _IteratedORTModule to handle it. handle_pool.append(module.register_forward_pre_hook(record_args)) for _, sub_module in module._modules.items():