Skip to content

Commit becf7cd

Browse files
committed
fix: update wrapper ref .__init__
1 parent f337814 commit becf7cd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

hubconf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,13 +17,13 @@
1717
for _optimizer in _get_supported_optimizers():
1818
name: str = _optimizer.__name__
1919
_func = _partial(_load_optimizer, optimizer=name)
20-
_update_wrapper(_func, _optimizer)
20+
_update_wrapper(_func, _optimizer.__init__)
2121
for n in (name, name.lower(), name.upper()):
2222
globals()[n] = _func
2323

2424
for _scheduler in _get_supported_lr_schedulers():
2525
name: str = _scheduler.__name__
2626
_func = _partial(_load_lr_scheduler, lr_scheduler=name)
27-
_update_wrapper(_func, _scheduler)
27+
_update_wrapper(_func, _scheduler.__init__)
2828
for n in (name, name.lower(), name.upper()):
2929
globals()[n] = _func

0 commit comments

Comments
 (0)