Skip to content

Commit f87162d

Browse files
committed
fixed formatting
Signed-off-by: mikail <mkhona@nvidia.com>
1 parent a9f16b0 commit f87162d

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

emerging_optimizers/riemannian_optimizers/normalized_optimizer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def __init__(
6464
)
6565
super().__init__(params, defaults)
6666

67-
@torch.no_grad()
67+
@torch.no_grad() # type: ignore[misc]
6868
def step(self, closure: Callable[[], float] | None = None) -> float | None:
6969
"""Performs a single optimization step.
7070
Args:
@@ -122,7 +122,7 @@ def __init__(
122122
weight_decay: float = 0.0,
123123
dim: int = 0,
124124
eps: float = 1e-8,
125-
correct_bias=True,
125+
correct_bias: bool = True,
126126
) -> None:
127127
"""An Adam-like optimizer for Normalized 2d Parameters
128128
@@ -153,7 +153,7 @@ def __init__(
153153
)
154154
super().__init__(params, defaults)
155155

156-
@torch.no_grad()
156+
@torch.no_grad() # type: ignore[misc]
157157
def step(self, closure: Callable[[], float] | None = None) -> float | None:
158158
"""Performs a single optimization step.
159159
Args:

0 commit comments

Comments
 (0)