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)