Mark "scaling Criteo" notebook to be skipped without TF (#537)

NVIDIA-Merlin · Aug 10, 2022 · e77f2fb · e77f2fb
1 parent 5d488d7
commit e77f2fb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/unit/examples/test_scaling_criteo_merlin_models.py b/tests/unit/examples/test_scaling_criteo_merlin_models.py
@@ -4,6 +4,8 @@
 from testbook import testbook
 from tests.conftest import REPO_ROOT
 
+pytest.importorskip("tensorflow")
+
 
 def test_func():
     with testbook(