From dfebffafb019a192d9d52be2c21368b130faea81 Mon Sep 17 00:00:00 2001
From: Stefan Nica <stefan@zenml.io>
Date: Mon, 7 Mar 2022 14:00:20 +0100
Subject: [PATCH] Add tensorboard example to integration tests

---
 examples/tensorboard/run.py        | 4 ++--
 tests/integration/test_examples.py | 7 +++++++
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/examples/tensorboard/run.py b/examples/tensorboard/run.py
index b92d4917377..15dd60f1876 100644
--- a/examples/tensorboard/run.py
+++ b/examples/tensorboard/run.py
@@ -19,7 +19,7 @@
 
 from rich import print
 
-from zenml.integrations.constants import AWS, TENSORFLOW
+from zenml.integrations.constants import TENSORFLOW
 from zenml.pipelines import pipeline
 from zenml.repository import Repository
 from zenml.steps import BaseStepConfig, Output, StepContext, step
@@ -107,7 +107,7 @@ def tf_evaluator(
 
 
 # Define the pipeline
-@pipeline(required_integrations=[TENSORFLOW, AWS], enable_cache=False)
+@pipeline(required_integrations=[TENSORFLOW], enable_cache=False)
 def mnist_pipeline(
     importer,
     normalizer,
diff --git a/tests/integration/test_examples.py b/tests/integration/test_examples.py
index aa7b3060711..93ff774fe80 100644
--- a/tests/integration/test_examples.py
+++ b/tests/integration/test_examples.py
@@ -172,6 +172,13 @@ class ExampleIntegrationTestConfiguration(NamedTuple):
             step_count=4,
         ),
     ),
+    ExampleIntegrationTestConfiguration(
+        name="tensorboard",
+        validation_function=generate_basic_validation_function(
+            pipeline_name="mnist_pipeline",
+            step_count=4,
+        ),
+    ),
 ]
 
 # flake8: noqa: C901