update docstring

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
vllm-project · Jan 29, 2025 · eb83e67 · eb83e67
1 parent 1b11b54
commit eb83e67
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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