Skip to content

Commit ac5daac

Browse files
committed
[skip ci] refactor: hubconf
1 parent e78e666 commit ac5daac

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

hubconf.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,9 @@
1212

1313
dependencies = ['torch']
1414

15-
if __name__ == '__main__':
16-
for optimizer in _get_supported_optimizers():
17-
name: str = optimizer.__name__
18-
for n in (name, name.lower(), name.upper()):
19-
func = _partial(_load_optimizer, optimizer=n)
20-
_update_wrapper(func, optimizer)
21-
globals()[n] = func
15+
for optimizer in _get_supported_optimizers():
16+
name: str = optimizer.__name__
17+
for n in (name, name.lower(), name.upper()):
18+
func = _partial(_load_optimizer, optimizer=n)
19+
_update_wrapper(func, optimizer)
20+
globals()[n] = func

0 commit comments

Comments
 (0)