From 39e804d0f8be8d14ef940f3a5ac71b09e3aa15d3 Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Tue, 23 Jul 2024 09:18:04 -0400 Subject: [PATCH] Use consistent param names in patch_extension(...) functions: context -> ctx. --- invokeai/backend/stable_diffusion/extensions/base.py | 2 +- invokeai/backend/stable_diffusion/extensions_manager.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/invokeai/backend/stable_diffusion/extensions/base.py b/invokeai/backend/stable_diffusion/extensions/base.py index 802af86e6d..835fe0aaf9 100644 --- a/invokeai/backend/stable_diffusion/extensions/base.py +++ b/invokeai/backend/stable_diffusion/extensions/base.py @@ -52,7 +52,7 @@ class ExtensionBase: return self._callbacks @contextmanager - def patch_extension(self, context: DenoiseContext): + def patch_extension(self, ctx: DenoiseContext): yield None @contextmanager diff --git a/invokeai/backend/stable_diffusion/extensions_manager.py b/invokeai/backend/stable_diffusion/extensions_manager.py index 1cae2e4219..a9e554ad98 100644 --- a/invokeai/backend/stable_diffusion/extensions_manager.py +++ b/invokeai/backend/stable_diffusion/extensions_manager.py @@ -52,13 +52,13 @@ class ExtensionsManager: cb.function(ctx) @contextmanager - def patch_extensions(self, context: DenoiseContext): + def patch_extensions(self, ctx: DenoiseContext): if self._is_canceled and self._is_canceled(): raise CanceledException with ExitStack() as exit_stack: for ext in self._extensions: - exit_stack.enter_context(ext.patch_extension(context)) + exit_stack.enter_context(ext.patch_extension(ctx)) yield None