Fix incorrect type hint
@@ -28,7 +28,7 @@ class NGD(torch.optim.Optimizer):
super().__init__(params, defaults=dict(lr=lr))
@torch.no_grad()
- def step(self) -> torch.Tensor:
+ def step(self) -> None:
"""Performs a single optimization step.
"""
for group in self.param_groups:
