fix: pass lint

metaopt · Jul 19, 2022 · a20dbb8 · a20dbb8
1 parent 7210c0d
commit a20dbb8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torchopt/_src/update.py b/torchopt/_src/update.py
@@ -35,7 +35,9 @@
 from torchopt._src import base
 
 
-def apply_updates(params: base.Params, updates: base.Updates, inplace: bool = True) -> base.Params:
+def apply_updates(
+    params: 'base.Params', updates: 'base.Updates', inplace: bool = True
+) -> 'base.Params':
     """Applies an update to the corresponding parameters.
 
     This is a utility functions that applies an update to a set of parameters,