diff --git a/tests/torchtune/modules/test_layer_dropout.py b/tests/torchtune/modules/test_layer_dropout.py index 73f42998d4..24494df91f 100644 --- a/tests/torchtune/modules/test_layer_dropout.py +++ b/tests/torchtune/modules/test_layer_dropout.py @@ -192,7 +192,7 @@ def test_get_scale_sin(self): assert_expected(actual_scale, expected_scale, atol=1e-7, rtol=1e-3) -class TestLayerDopoutModel: +class TestLayerDropoutModel: def test_prepare_layer_dropout_uniform(self): class MockModel(torch.nn.Module): def __init__(self):