fix: [pre-commit.ci] auto fixes [...]

metaopt · Nov 4, 2024 · 97b8058 · 97b8058
1 parent 56b0abd
commit 97b8058
Show file tree

Hide file tree

Showing 7 changed files with 5,550 additions and 2,913 deletions.
diff --git a/torchopt/nn/stateless.py b/torchopt/nn/stateless.py
@@ -84,7 +84,7 @@ def reparametrize(
     module: nn.Module,
     named_tensors: dict[str, torch.Tensor] | Iterable[tuple[str, torch.Tensor]],
     allow_missing: bool = False,
-) -> Generator[nn.Module, None, None]:
+) -> Generator[nn.Module]:
     """Reparameterize the module parameters and/or buffers."""
     if not isinstance(named_tensors, dict):
         named_tensors = dict(named_tensors)