From b6ba6cb7c90372cb5462855c21e5f52fc9130557 Mon Sep 17 00:00:00 2001 From: Jason Gross Date: Wed, 20 Mar 2024 03:23:10 -0700 Subject: [PATCH] fix outdated copy/paste names in test_l1 (#203) --- .../metrics/loss/tests/test_l1_absolute_loss.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sparse_autoencoder/metrics/loss/tests/test_l1_absolute_loss.py b/sparse_autoencoder/metrics/loss/tests/test_l1_absolute_loss.py index 4a2b7485..ca6435ca 100644 --- a/sparse_autoencoder/metrics/loss/tests/test_l1_absolute_loss.py +++ b/sparse_autoencoder/metrics/loss/tests/test_l1_absolute_loss.py @@ -38,13 +38,13 @@ ), ], ) -def test_l2_reconstruction_loss( +def test_l1_absolute_loss( learned_activations: Float[ Tensor, Axis.names(Axis.BATCH, Axis.COMPONENT_OPTIONAL, Axis.LEARNT_FEATURE) ], expected_loss: Float[Tensor, Axis.COMPONENT_OPTIONAL], ) -> None: - """Test the L2 reconstruction loss.""" + """Test the L1 absolute loss.""" num_components: int = learned_activations.shape[1] if learned_activations.ndim == 3 else 1 # noqa: PLR2004 l1 = L1AbsoluteLoss(num_components)