diff --git a/timm/__init__.py b/timm/__init__.py
index 3d38cdb97..0e48cf7ce 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",
+]