diff --git a/evals/unlearning/main.py b/evals/unlearning/main.py index 66161f8..aece340 100644 --- a/evals/unlearning/main.py +++ b/evals/unlearning/main.py @@ -147,7 +147,7 @@ def run_eval( if "topk" in sae_name: assert isinstance(sae.activation_fn, TopK) - single_sae_eval_results = run_eval_single_sae(model, sae, sae_name, config) + run_eval_single_sae(model, sae, sae_name, config) sae_folder = os.path.join("results/metrics", sae_name)