Apply suggestion from @kylesayrs

Co-authored-by: Kyle Sayers <kylesayrs@gmail.com>
Signed-off-by: Brian Dellabetta <brian-dellabetta@users.noreply.github.com>

Author: brian-dellabetta

src/compressed_tensors/utils/helpers.py

@@ -391,19 +391,10 @@ def patch_attrs(bases: Iterable[Any], attr: str, values: Iterable[Any]):
     >>> assert not hasattr(obj1, "attribute")
     >>> assert not hasattr(obj2, "attribute")
     """
-    _sentinel = object()
-    original_values = [getattr(base, attr, _sentinel) for base in bases]
-
-    for base, value in zip(bases, values):
-        setattr(base, attr, value)
-    try:
+    with contextlib.exitstack() as stack:
+        for base, value in zip(bases, values):
+            stack.add(patch_attr(base, attr, value))
         yield
-    finally:
-        for base, original_value in zip(bases, original_values):
-            if original_value is not _sentinel:
-                setattr(base, attr, original_value)
-            else:
-                delattr(base, attr)
 
 
 class ParameterizedDefaultDict(dict):