diff --git a/python/mlx/nn/layers/dropout.py b/python/mlx/nn/layers/dropout.py index 8c7326e2d..f656f8db3 100644 --- a/python/mlx/nn/layers/dropout.py +++ b/python/mlx/nn/layers/dropout.py @@ -19,7 +19,7 @@ class Dropout(Module): super().__init__() if p < 0 or p >= 1: - raise ValueError("The dropout probability {p} is not in [0, 1)") + raise ValueError(f"The dropout probability {p} is not in [0, 1)") self._p_1 = 1 - p