From 0b26abf65b2bb18071464573aa0629f84696017e Mon Sep 17 00:00:00 2001 From: Younghwan Na Date: Fri, 13 May 2022 08:54:41 +0900 Subject: [PATCH] Rename congtext variable --- mmtune/mm/context/manager.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/mmtune/mm/context/manager.py b/mmtune/mm/context/manager.py index 71ef7cbc..e6528192 100644 --- a/mmtune/mm/context/manager.py +++ b/mmtune/mm/context/manager.py @@ -28,13 +28,13 @@ def __init__(self, def __call__(self, func): - def inner(*tpargs, **kwargs): - kwargs['searched_cfg'] = tpargs[0] - kwargs['base_cfg'] = self.base_cfg - kwargs['args'] = self.args + def inner(*searched_cfg, **context): + context['searched_cfg'] = searched_cfg[0] + context['base_cfg'] = self.base_cfg + context['args'] = self.args for rewriter in self.rewriters: - kwargs = rewriter(kwargs) - return func(**kwargs) + context = rewriter(context) + return func(**context) return inner