Skip to content

Commit

Permalink
fix formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
cuent committed Jun 4, 2020
1 parent 530cede commit cb0d5e0
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/metrics/functional/test_classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -272,7 +272,8 @@ def test_auroc(pred, target, expected):
score = auroc(pred, target).item()
assert score == expected

@pytest.mark.parametrize(['x', 'y', 'expected'],[

@pytest.mark.parametrize(['x', 'y', 'expected'], [
pytest.param(torch.tensor([0, 1]), torch.tensor([0, 1]), 0.5),
pytest.param(torch.tensor([1, 0]), torch.tensor([0, 1]), 0.5),
pytest.param(torch.tensor([1, 0, 0]), torch.tensor([0, 1, 1]), 0.5),
Expand Down

0 comments on commit cb0d5e0

Please sign in to comment.