Fix misc flake8 errors

Signed-off-by: Fynn Schmitt-Ulms <[email protected]>

Author: fynnsu

src/compressed_tensors/utils/offload.py

@@ -282,7 +282,6 @@ def disable_hf_hook(module: torch.nn.Module):
     hooks = {}
 
     def collect_hooks(module):
-        nonlocal hooks
         if hasattr(module, "_hf_hook"):
             hooks[module] = module._hf_hook
             remove_hook_from_module(module)

tests/test_examples/test_bitmask_compression_ipynb.py

@@ -16,7 +16,7 @@
 
 
 nbformat = pytest.importorskip("nbformat")
-from nbconvert.preprocessors import ExecutePreprocessor
+from nbconvert.preprocessors import ExecutePreprocessor  # noqa: E402
 
 
 @pytest.mark.skip(

tests/test_transform/test_transform_config.py

@@ -66,6 +66,6 @@ def test_multiple_groups():
         type="hadamard",
         apply=[linear_args_2],
     )
-    config = TransformConfig(
+    _ = TransformConfig(
         config_groups={"transform_0": scheme_1, "transform_1": scheme_2}
     )

tests/test_utils/test_offload.py

@@ -441,8 +441,8 @@ def test_delete_offload_module(exec_device):
     register_offload_module(model.linear, "child", child)
     delete_offload_module(model, "child")
     delete_offload_module(model.linear, "child")
-    assert not child in model.children()
-    assert not child in model.linear.children()
+    assert child not in model.children()
+    assert child not in model.linear.children()
 
     # with offloading
     model = ExampleModel()
@@ -452,8 +452,8 @@ def test_delete_offload_module(exec_device):
     register_offload_module(model.linear, "child", child)
     delete_offload_module(model, "child")
     delete_offload_module(model.linear, "child")
-    assert not child in model.children()
-    assert not child in model.linear.children()
+    assert child not in model.children()
+    assert child not in model.linear.children()
 
 
 @requires_gpu