Add error handling for log_dashboard_urls importstatements

Author: htahir1

pipelines/cache_pipeline/cache_pipeline.py

@@ -4,7 +4,10 @@
 from zenml import pipeline, step
 from zenml.logger import get_logger
 
-from utils import log_dashboard_urls  # type: ignore
+try:
+    from utils import log_dashboard_urls  # type: ignore
+except ImportError:
+    log_dashboard_urls = lambda name: print(f"📊 Pipeline '{name}' completed!")
 
 logger = get_logger(__name__)
 

pipelines/fan_pipeline/fan_pipeline.py

@@ -3,7 +3,10 @@
 from zenml.client import Client
 from zenml.logger import get_logger
 
-from utils import log_dashboard_urls  # type: ignore
+try:
+    from utils import log_dashboard_urls  # type: ignore
+except ImportError:
+    log_dashboard_urls = lambda name: print(f"📊 Pipeline '{name}' completed!")
 
 logger = get_logger(__name__)
 

pipelines/hello_pipeline/hello_pipeline.py

@@ -1,7 +1,10 @@
 from zenml import pipeline, step
 from zenml.logger import get_logger
 
-from utils import log_dashboard_urls  # type: ignore
+try:
+    from utils import log_dashboard_urls  # type: ignore
+except ImportError:
+    log_dashboard_urls = lambda name: print(f"📊 Pipeline '{name}' completed!")
 
 logger = get_logger(__name__)
 

pipelines/io_pipeline/io_pipeline.py

@@ -4,7 +4,10 @@
 from zenml import pipeline, step
 from zenml.logger import get_logger
 
-from utils import log_dashboard_urls  # type: ignore
+try:
+    from utils import log_dashboard_urls  # type: ignore
+except ImportError:
+    log_dashboard_urls = lambda name: print(f"📊 Pipeline '{name}' completed!")
 
 logger = get_logger(__name__)
 

pipelines/meta_pipeline/meta_pipeline.py

@@ -2,7 +2,10 @@
 from zenml import log_metadata, pipeline, step
 from zenml.logger import get_logger
 
-from utils import log_dashboard_urls  # type: ignore
+try:
+    from utils import log_dashboard_urls  # type: ignore
+except ImportError:
+    log_dashboard_urls = lambda name: print(f"📊 Pipeline '{name}' completed!")
 
 logger = get_logger(__name__)
 

pipelines/param_pipeline/param_pipeline.py

@@ -2,7 +2,10 @@
 from zenml import pipeline, step
 from zenml.logger import get_logger
 
-from utils import log_dashboard_urls  # type: ignore
+try:
+    from utils import log_dashboard_urls  # type: ignore
+except ImportError:
+    log_dashboard_urls = lambda name: print(f"📊 Pipeline '{name}' completed!")
 
 logger = get_logger(__name__)
 

pipelines/robust_pipeline/robust_pipeline.py

@@ -6,7 +6,10 @@
 from zenml.config import StepRetryConfig
 from zenml.logger import get_logger
 
-from utils import log_dashboard_urls
+try:
+    from utils import log_dashboard_urls  # type: ignore
+except ImportError:
+    log_dashboard_urls = lambda name: print(f"📊 Pipeline '{name}' completed!")
 
 logger = get_logger(__name__)
 

pipelines/tagged_pipeline/tagged_pipeline.py

@@ -4,7 +4,10 @@
 from zenml import ArtifactConfig, Tag, add_tags, pipeline, step
 from zenml.logger import get_logger
 
-from utils import log_dashboard_urls  # type: ignore
+try:
+    from utils import log_dashboard_urls  # type: ignore
+except ImportError:
+    log_dashboard_urls = lambda name: print(f"📊 Pipeline '{name}' completed!")
 
 logger = get_logger(__name__)
 

pipelines/viz_pipeline/viz_pipeline.py

@@ -8,7 +8,10 @@
 from zenml.logger import get_logger
 from zenml.types import HTMLString
 
-from utils import log_dashboard_urls  # type: ignore
+try:
+    from utils import log_dashboard_urls  # type: ignore
+except ImportError:
+    log_dashboard_urls = lambda name: print(f"📊 Pipeline '{name}' completed!")
 
 logger = get_logger(__name__)
 

pipelines/yaml_pipeline/yaml_pipeline.py

@@ -2,7 +2,10 @@
 from zenml import pipeline, step
 from zenml.logger import get_logger
 
-from utils import log_dashboard_urls  # type: ignore
+try:
+    from utils import log_dashboard_urls  # type: ignore
+except ImportError:
+    log_dashboard_urls = lambda name: print(f"📊 Pipeline '{name}' completed!")
 
 logger = get_logger(__name__)