diff --git a/torch/optim/optimizer.py b/torch/optim/optimizer.py index 8cced80c084d..e5d989274f80 100644 --- a/torch/optim/optimizer.py +++ b/torch/optim/optimizer.py @@ -1004,7 +1004,7 @@ class Optimizer: torch._foreach_zero_(grads) @overload - def step(self, closure: None = ...) -> None: + def step(self, closure: None = None) -> None: ... @overload