Dynamically import entrypoint in root __init__

Author: schustmi

src/zenml/__init__.py

@@ -14,6 +14,7 @@
 """Initialization for ZenML."""
 
 import os
+from typing import Any
 
 ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
 
@@ -26,6 +27,16 @@
 
 init_logging()
 
+def __getattr__(name: str) -> Any:
+    # We allow directly accessing the entrypoint module as `zenml.entrypoint`
+    # as this is needed for some orchestrators. Instead of directly importing
+    # the entrypoint module here, we import it dynamically. This avoids a
+    # warning when running the `zenml.entrypoints.entrypoint` module directly.
+    if name == "entrypoint":
+        from zenml.entrypoints import entrypoint
+        return entrypoint
+
+    raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Need to import zenml.models before zenml.config to avoid circular imports
 from zenml.models import *  # noqa: F401
@@ -50,7 +61,7 @@
 from zenml.steps.utils import log_step_metadata
 from zenml.utils.metadata_utils import log_metadata
 from zenml.utils.tag_utils import Tag, add_tags, remove_tags
-from zenml.entrypoints import entrypoint
+
 
 __all__ = [
     "add_tags",
@@ -72,5 +83,4 @@
     "register_artifact",
     "show",
     "step",
-    "entrypoint",
 ]