From eb83e67f284ad9c58e7477a441779a956046afc6 Mon Sep 17 00:00:00 2001 From: Kyle Sayers Date: Wed, 29 Jan 2025 05:18:56 +0000 Subject: [PATCH] update docstring Signed-off-by: Kyle Sayers --- src/llmcompressor/modifiers/utils/hooks.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/llmcompressor/modifiers/utils/hooks.py b/src/llmcompressor/modifiers/utils/hooks.py index b35dc7cac..386d58cac 100644 --- a/src/llmcompressor/modifiers/utils/hooks.py +++ b/src/llmcompressor/modifiers/utils/hooks.py @@ -76,7 +76,12 @@ def wrapped_hook(*args, **kwargs): return handle def remove_hooks(self, handles: Optional[Set[RemovableHandle]] = None): - """Remove all hooks belonging to a modifier""" + """ + Removes hooks registered by this modifier + + :param handles: optional list of handles to remove, defaults to all hooks + registerd by this modifier + """ if handles is None: handles = self._hooks