Use import alias

huggingface · Jan 22, 2025 · d81da93 · d81da93
1 parent 4de1abf
commit d81da93
Showing 1 changed file with 16 additions and 28 deletions.
diff --git 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",
-]