Add tensorboard example to integration tests

zenml-io · Mar 7, 2022 · dfebffa · dfebffa
1 parent 2806dc5
commit dfebffa
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git 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
@@ -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