Move tests

etowahadams · Nov 2, 2024 · 3447cc3 · 3447cc3
1 parent f86f947
commit 3447cc3
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/...ic_regression_probe/test_single_latent.py → ...ic_regression_probe/test_single_latent.py b/...ic_regression_probe/test_single_latent.py → ...ic_regression_probe/test_single_latent.py
@@ -14,9 +14,7 @@ class TestSingleLatentProbe(unittest.TestCase):
         "interprot.logistic_regression_probe.single_latent.prepare_arrays_for_logistic_regression"
     )
     @patch("interprot.logistic_regression_probe.single_latent.torch.load")
-    @patch(
-        "interprot.logistic_regression_probe.single_latent.AutoTokenizer.from_pretrained"
-    )
+    @patch("interprot.logistic_regression_probe.single_latent.AutoTokenizer.from_pretrained")
     @patch("interprot.logistic_regression_probe.single_latent.EsmModel.from_pretrained")
     @patch("interprot.logistic_regression_probe.single_latent.SparseAutoencoder")
     def test_single_latent_e2e(

diff --git a/...s/logistic_regression_probe/test_utils.py → ...s/logistic_regression_probe/test_utils.py b/...s/logistic_regression_probe/test_utils.py → ...s/logistic_regression_probe/test_utils.py
diff --git a/viz/src/SAEConfigs.ts b/viz/src/SAEConfigs.ts
@@ -42,7 +42,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
       {
         name: "kinase helix",
         dim: 594,
-        desc: "This feature activates strongly on a specific helix in kinase domains and weakly on surrounding beta strands. The highlighted helix is always opposing a beta sheet.",
+        desc: "This feature activates strongly on a specific helix in kinase domains and weakly on surrounding beta strands. The highlighted helix is always the one opposed to the beta sheet.",
         group: "structural",
       },
       {