[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

README.md

@@ -230,7 +230,7 @@ The experiments discussed in [the blog post](https://blog.eleuther.ai/autointerp
 
 ## Development
 
-Run unit tests: 
+Run unit tests:
 
 ```pytest .```
 

delphi/__main__.py

@@ -6,7 +6,6 @@
 
 import orjson
 import torch
-from simple_parsing import ArgumentParser
 from datasets import Dataset
 from simple_parsing import ArgumentParser
 from sparsify.data import chunk_and_tokenize
@@ -28,10 +27,9 @@
 from delphi.log.result_analysis import log_results
 from delphi.pipeline import Pipe, Pipeline, process_wrapper
 from delphi.scorers import DetectionScorer, FuzzingScorer
+from delphi.semantic_index.index import build_or_load_index, load_index
 from delphi.sparse_coders import load_hooks_sparse_coders, load_sparse_coders
 from delphi.utils import load_tokenized_data
-from delphi.sparse_coders import load_sparse_coders
-from delphi.semantic_index.index import build_or_load_index, load_index
 
 
 def load_artifacts(run_cfg: RunConfig):
@@ -184,12 +182,14 @@ def explainer_postprocess(result):
         )
         postprocess = None
     else:
-        explainer = DefaultExplainer(
-            client,
-            threshold=0.3,
-            verbose=run_cfg.verbose,
-        ),
-        postprocess= explainer_postprocess
+        explainer = (
+            DefaultExplainer(
+                client,
+                threshold=0.3,
+                verbose=run_cfg.verbose,
+            ),
+        )
+        postprocess = explainer_postprocess
 
     explainer_pipe = Pipe(
         process_wrapper(

delphi/tests/conftest.py

@@ -77,9 +77,7 @@ def cache_setup(tmp_path_factory, mock_dataset: torch.Tensor, model: PreTrainedM
         sparse_model="EleutherAI/sae-pythia-160m-32k",
         hookpoints=["layers.1"],
     )
-    hookpoint_to_sparse_encode, _ = load_hooks_sparse_coders(
-        model, run_cfg_gemma
-    )
+    hookpoint_to_sparse_encode, _ = load_hooks_sparse_coders(model, run_cfg_gemma)
     print(hookpoint_to_sparse_encode)
     # Define cache config and initialize cache
     cache = LatentCache(

delphi/tests/test_autoencoders/test_sparse_coders.py

@@ -71,7 +71,9 @@ def test_retrieve_autoencoders_from_sparsify(dummy_model, run_cfg_sparsify):
     """
     Tests that load_hooks_sparse_coders retrieves autoencoders from Sparsify.
     """
-    hookpoint_to_sparse_encode,_ = load_hooks_sparse_coders(dummy_model, run_cfg_sparsify)
+    hookpoint_to_sparse_encode, _ = load_hooks_sparse_coders(
+        dummy_model, run_cfg_sparsify
+    )
     # Verify that we received a dictionary of autoencoders.
     assert (
         isinstance(hookpoint_to_sparse_encode, dict)
@@ -94,7 +96,9 @@ def test_retrieve_autoencoders_from_gemma(dummy_model, run_cfg_gemma):
     """
     Tests that load_hooks_sparse_coders retrieves autoencoders from Gemma.
     """
-    hookpoint_to_sparse_encode,transcode = load_hooks_sparse_coders(dummy_model, run_cfg_gemma)
+    hookpoint_to_sparse_encode, transcode = load_hooks_sparse_coders(
+        dummy_model, run_cfg_gemma
+    )
     # Verify that we received a dictionary of autoencoders.
     assert (
         isinstance(hookpoint_to_sparse_encode, dict)