From a20dbb841b2ca31dd7658d918f473da8eb8cf001 Mon Sep 17 00:00:00 2001 From: Benjamin-eecs Date: Tue, 19 Jul 2022 18:19:44 +0800 Subject: [PATCH] fix: pass lint --- torchopt/_src/update.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/torchopt/_src/update.py b/torchopt/_src/update.py index 56e509401..5628b0cb0 100644 --- 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,