From 3447cc3f8167899d52a1f895e3538a69934f93f2 Mon Sep 17 00:00:00 2001 From: Liam Date: Sat, 2 Nov 2024 10:00:28 -0400 Subject: [PATCH] Move tests --- .../tests}/logistic_regression_probe/test_single_latent.py | 4 +--- .../tests}/logistic_regression_probe/test_utils.py | 0 viz/src/SAEConfigs.ts | 2 +- 3 files changed, 2 insertions(+), 4 deletions(-) rename {tests => interprot/tests}/logistic_regression_probe/test_single_latent.py (96%) rename {tests => interprot/tests}/logistic_regression_probe/test_utils.py (100%) diff --git a/tests/logistic_regression_probe/test_single_latent.py b/interprot/tests/logistic_regression_probe/test_single_latent.py similarity index 96% rename from tests/logistic_regression_probe/test_single_latent.py rename to interprot/tests/logistic_regression_probe/test_single_latent.py index 3c01830..b46e0df 100644 --- a/tests/logistic_regression_probe/test_single_latent.py +++ b/interprot/tests/logistic_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/tests/logistic_regression_probe/test_utils.py b/interprot/tests/logistic_regression_probe/test_utils.py similarity index 100% rename from tests/logistic_regression_probe/test_utils.py rename to interprot/tests/logistic_regression_probe/test_utils.py diff --git a/viz/src/SAEConfigs.ts b/viz/src/SAEConfigs.ts index 91d0a21..fa85275 100644 --- a/viz/src/SAEConfigs.ts +++ b/viz/src/SAEConfigs.ts @@ -42,7 +42,7 @@ export const SAE_CONFIGS: Record = { { 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", }, {