Skip to content

Commit

Permalink
fix outdated copy/paste names in test_l1 (ai-safety-foundation#203)
Browse files Browse the repository at this point in the history
  • Loading branch information
JasonGross authored Mar 20, 2024
1 parent 8dc963a commit b6ba6cb
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down

0 comments on commit b6ba6cb

Please sign in to comment.