From faed62171b0bf582329d5589c7613c8d8b481589 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Sat, 11 Jan 2025 14:31:41 +0100 Subject: [PATCH 1/2] timm: add __all__ to __init__ --- timm/__init__.py | 30 ++++++++++++++++++++++++++++-- 1 file changed, 28 insertions(+), 2 deletions(-) diff --git a/timm/__init__.py b/timm/__init__.py index 3d38cdb978..0e48cf7ce9 100644 --- a/timm/__init__.py +++ b/timm/__init__.py @@ -1,4 +1,30 @@ from .version import __version__ from .layers import is_scriptable, is_exportable, set_scriptable, set_exportable -from .models import create_model, list_models, list_pretrained, is_model, list_modules, model_entrypoint, \ - is_model_pretrained, get_pretrained_cfg, get_pretrained_cfg_value +from .models import ( + create_model, + list_models, + list_pretrained, + is_model, + list_modules, + model_entrypoint, + is_model_pretrained, + get_pretrained_cfg, + get_pretrained_cfg_value, +) + +__all__ = [ + "__version__", + "is_scriptable", + "is_exportable", + "set_scriptable", + "set_exportable", + "create_model", + "list_models", + "list_pretrained", + "is_model", + "list_modules", + "model_entrypoint", + "is_model_pretrained", + "get_pretrained_cfg", + "get_pretrained_cfg_value", +] From 1e189be46d608d6d20d35d2a4b66e40787fd574c Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Wed, 22 Jan 2025 10:08:18 +0100 Subject: [PATCH 2/2] Use import alias --- timm/__init__.py | 44 ++++++++++++++++---------------------------- 1 file changed, 16 insertions(+), 28 deletions(-) diff --git a/timm/__init__.py b/timm/__init__.py index 0e48cf7ce9..229379ec98 100644 --- a/timm/__init__.py +++ b/timm/__init__.py @@ -1,30 +1,18 @@ -from .version import __version__ -from .layers import is_scriptable, is_exportable, set_scriptable, set_exportable +from .version import __version__ as __version__ +from .layers import ( + is_scriptable as is_scriptable, + is_exportable as is_exportable, + set_scriptable as set_scriptable, + set_exportable as set_exportable, +) from .models import ( - create_model, - list_models, - list_pretrained, - is_model, - list_modules, - model_entrypoint, - is_model_pretrained, - get_pretrained_cfg, - get_pretrained_cfg_value, + create_model as create_model, + list_models as list_models, + list_pretrained as list_pretrained, + is_model as is_model, + list_modules as list_modules, + model_entrypoint as model_entrypoint, + is_model_pretrained as is_model_pretrained, + get_pretrained_cfg as get_pretrained_cfg, + get_pretrained_cfg_value as get_pretrained_cfg_value, ) - -__all__ = [ - "__version__", - "is_scriptable", - "is_exportable", - "set_scriptable", - "set_exportable", - "create_model", - "list_models", - "list_pretrained", - "is_model", - "list_modules", - "model_entrypoint", - "is_model_pretrained", - "get_pretrained_cfg", - "get_pretrained_cfg_value", -]