Skip to content

Commit 7c17b14

Browse files
committed
fix: use update_wrapper
1 parent 74caee6 commit 7c17b14

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

hubconf.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,15 @@
11
dependencies = ["torch"]
22

3-
from functools import partial
3+
from functools import partial as _partial, update_wrapper as _update_wrapper
44

5-
from pytorch_optimizer import get_supported_optimizers, load_optimizer
5+
from pytorch_optimizer import (
6+
get_supported_optimizers as _get_supported_optimizers,
7+
load_optimizer as _load_optimizer,
8+
)
69

7-
for optimizer in get_supported_optimizers():
10+
for optimizer in _get_supported_optimizers():
811
name = optimizer.__name__
912
for n in (name, name.lower()):
10-
globals()[n] = partial(load_optimizer, optimizer=n)
13+
func = _partial(_load_optimizer, optimizer=n)
14+
_update_wrapper(func, optimizer)
15+
globals()[n] = func

0 commit comments

Comments
 (0)