Make test file names consistent

adamkarvonen · Nov 16, 2024 · 70e2eaa · 70e2eaa
1 parent 4ed9602
commit 70e2eaa
Show file tree

Hide file tree

Showing 8 changed files with 4 additions and 10 deletions.
diff --git 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/...eduped_layer_4_expected_eval_results.json → ...tointerp/autointerp_expected_results.json b/...eduped_layer_4_expected_eval_results.json → ...tointerp/autointerp_expected_results.json
diff --git a/...k_resid_post__trainer_2_eval_results.json → ...s/test_data/mdl/mdl_expected_results.json b/...k_resid_post__trainer_2_eval_results.json → ...s/test_data/mdl/mdl_expected_results.json
diff --git a/...ed_scr_layer_4_expected_eval_results.json → ...a/shift_and_tpp/scr_expected_results.json b/...ed_scr_layer_4_expected_eval_results.json → ...a/shift_and_tpp/scr_expected_results.json
diff --git a/...ed_tpp_layer_4_expected_eval_results.json → ...a/shift_and_tpp/tpp_expected_results.json b/...ed_tpp_layer_4_expected_eval_results.json → ...a/shift_and_tpp/tpp_expected_results.json
diff --git a/...k_resid_post__trainer_2_eval_results.json → ...learning/unlearning_expected_results.json b/...k_resid_post__trainer_2_eval_results.json → ...learning/unlearning_expected_results.json
diff --git 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
@@ -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():