Skip to content

Commit 4de1abf

Browse files
adamjstewartrwightman
authored andcommitted
timm: add __all__ to __init__
1 parent bda46f8 commit 4de1abf

File tree

1 file changed

+28
-2
lines changed

1 file changed

+28
-2
lines changed

Diff for: timm/__init__.py

+28-2
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,30 @@
11
from .version import __version__
22
from .layers import is_scriptable, is_exportable, set_scriptable, set_exportable
3-
from .models import create_model, list_models, list_pretrained, is_model, list_modules, model_entrypoint, \
4-
is_model_pretrained, get_pretrained_cfg, get_pretrained_cfg_value
3+
from .models import (
4+
create_model,
5+
list_models,
6+
list_pretrained,
7+
is_model,
8+
list_modules,
9+
model_entrypoint,
10+
is_model_pretrained,
11+
get_pretrained_cfg,
12+
get_pretrained_cfg_value,
13+
)
14+
15+
__all__ = [
16+
"__version__",
17+
"is_scriptable",
18+
"is_exportable",
19+
"set_scriptable",
20+
"set_exportable",
21+
"create_model",
22+
"list_models",
23+
"list_pretrained",
24+
"is_model",
25+
"list_modules",
26+
"model_entrypoint",
27+
"is_model_pretrained",
28+
"get_pretrained_cfg",
29+
"get_pretrained_cfg_value",
30+
]

0 commit comments

Comments
 (0)