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