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