Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
csteinmetz1 committed Apr 19, 2023
1 parent e0502e1 commit 9e99098
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions tests/test_auraloss.py
Original file line number Diff line number Diff line change
Expand Up @@ -126,7 +126,6 @@ def test_multires_mel():
assert res is not None



def test_perceptual_multires_mel():
target = torch.rand(8, 2, 44100)
pred = torch.rand(8, 2, 44100)
Expand All @@ -143,6 +142,7 @@ def test_perceptual_multires_mel():
res = loss(pred, target)
assert res is not None


def test_stft_l2():
N = 32
n = torch.arange(N)
Expand Down Expand Up @@ -196,4 +196,3 @@ def test_multires_l2():
expected_loss = ((N // 2) ** 2) / (N // 2 + 1)

torch.testing.assert_close(res, torch.tensor(expected_loss), rtol=1e-3, atol=1e-3)

0 comments on commit 9e99098

Please sign in to comment.