diff --git a/tests/test_autointerp.py b/tests/test_autointerp.py index c76c2c4..6643732 100644 --- a/tests/test_autointerp.py +++ b/tests/test_autointerp.py @@ -5,9 +5,7 @@ import sae_bench_utils.testing_utils as testing_utils from sae_bench_utils.sae_selection_utils import select_saes_multiple_patterns -results_filename = ( - "tests/test_data/autointerp/pythia-70m-deduped_layer_4_expected_eval_results.json" -) +results_filename = "tests/test_data/autointerp/autointerp_expected_results.json" def test_end_to_end_different_seed(): diff --git a/tests/test_data/autointerp/pythia-70m-deduped_layer_4_expected_eval_results.json b/tests/test_data/autointerp/autointerp_expected_results.json similarity index 100% rename from tests/test_data/autointerp/pythia-70m-deduped_layer_4_expected_eval_results.json rename to tests/test_data/autointerp/autointerp_expected_results.json diff --git a/tests/test_data/mdl/sae_bench_pythia70m_sweep_topk_ctx128_0730_blocks.4.hook_resid_post__trainer_2_eval_results.json b/tests/test_data/mdl/mdl_expected_results.json similarity index 100% rename from tests/test_data/mdl/sae_bench_pythia70m_sweep_topk_ctx128_0730_blocks.4.hook_resid_post__trainer_2_eval_results.json rename to tests/test_data/mdl/mdl_expected_results.json diff --git a/tests/test_data/shift_and_tpp/pythia-70m-deduped_scr_layer_4_expected_eval_results.json b/tests/test_data/shift_and_tpp/scr_expected_results.json similarity index 100% rename from tests/test_data/shift_and_tpp/pythia-70m-deduped_scr_layer_4_expected_eval_results.json rename to tests/test_data/shift_and_tpp/scr_expected_results.json diff --git a/tests/test_data/shift_and_tpp/pythia-70m-deduped_tpp_layer_4_expected_eval_results.json b/tests/test_data/shift_and_tpp/tpp_expected_results.json similarity index 100% rename from tests/test_data/shift_and_tpp/pythia-70m-deduped_tpp_layer_4_expected_eval_results.json rename to tests/test_data/shift_and_tpp/tpp_expected_results.json diff --git a/tests/test_data/unlearning/sae_bench_gemma-2-2b_sweep_topk_ctx128_ef8_0824_blocks.3.hook_resid_post__trainer_2_eval_results.json b/tests/test_data/unlearning/unlearning_expected_results.json similarity index 100% rename from tests/test_data/unlearning/sae_bench_gemma-2-2b_sweep_topk_ctx128_ef8_0824_blocks.3.hook_resid_post__trainer_2_eval_results.json rename to tests/test_data/unlearning/unlearning_expected_results.json diff --git a/tests/test_shift_and_tpp.py b/tests/test_shift_and_tpp.py index 0f06eda..1023f54 100644 --- a/tests/test_shift_and_tpp.py +++ b/tests/test_shift_and_tpp.py @@ -5,12 +5,8 @@ import sae_bench_utils.testing_utils as testing_utils from sae_bench_utils.sae_selection_utils import select_saes_multiple_patterns -tpp_results_filename = ( - "tests/test_data/shift_and_tpp/pythia-70m-deduped_tpp_layer_4_expected_eval_results.json" -) -scr_results_filename = ( - "tests/test_data/shift_and_tpp/pythia-70m-deduped_scr_layer_4_expected_eval_results.json" -) +tpp_results_filename = "tests/test_data/shift_and_tpp/tpp_expected_results.json" +scr_results_filename = "tests/test_data/shift_and_tpp/scr_expected_results.json" def test_scr_end_to_end_different_seed(): diff --git a/tests/test_unlearning.py b/tests/test_unlearning.py index 5222131..3e00731 100644 --- a/tests/test_unlearning.py +++ b/tests/test_unlearning.py @@ -5,7 +5,7 @@ import sae_bench_utils.testing_utils as testing_utils from sae_bench_utils.sae_selection_utils import select_saes_multiple_patterns -results_filename = "tests/test_data/unlearning/sae_bench_gemma-2-2b_sweep_topk_ctx128_ef8_0824_blocks.3.hook_resid_post__trainer_2_eval_results.json" +results_filename = "tests/test_data/unlearning/unlearning_expected_results.json" def test_end_to_end_different_seed():