[skip ci] refactor: type hint for OPTIMIZER_LIST

kozistr · kozistr · commit d9ff347e8b8d · 2022-09-01T20:57:48.000+09:00
diff --git a/pytorch_optimizer/__init__.py b/pytorch_optimizer/__init__.py
@@ -36,7 +36,7 @@
     unit_norm,
 )
 
-OPTIMIZER_LIST: List = [
+OPTIMIZER_LIST: List[Type[Optimizer]] = [
     AdaBelief,
     AdaBound,
     AdamP,

Original file line number	Diff line number	Diff line change
`@@ -36,7 +36,7 @@`
`36`	`36`	`unit_norm,`
`37`	`37`	`)`
`38`	`38`
`39`		`-OPTIMIZER_LIST: List = [`
	`39`	`+OPTIMIZER_LIST: List[Type[Optimizer]] = [`
`40`	`40`	`AdaBelief,`
`41`	`41`	`AdaBound,`
`42`	`42`	`AdamP,`