Skip to content

Commit

Permalink
Update tests/tests_fabric/plugins/precision/test_amp.py
Browse files Browse the repository at this point in the history
  • Loading branch information
awaelchli authored May 29, 2023
1 parent a8806c5 commit bb7ae8c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/tests_fabric/plugins/precision/test_amp.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ def test_amp_precision_optimizer_step_without_scaler():
optimizer.step.assert_called_once_with(keyword="arg")


def test_amp_precisionparameter_validation():
def test_amp_precision_parameter_validation():
MixedPrecision("16-mixed", "cpu") # should not raise exception
MixedPrecision("bf16-mixed", "cpu")

Expand Down

0 comments on commit bb7ae8c

Please sign in to comment.