adding import or skip for tensorflow framework required by examples t…

…ests (#1691)
NVIDIA-Merlin · Oct 8, 2022 · 8f87c34 · 8f87c34
1 parent b2144b4
commit 8f87c34
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/tests/unit/examples/test_01-Getting-started.py b/tests/unit/examples/test_01-Getting-started.py
@@ -14,10 +14,12 @@
 # limitations under the License.
 #
 import nest_asyncio
+import pytest
 from testbook import testbook
 
 from tests.conftest import REPO_ROOT
 
+pytest.importorskip("tensorflow")
 nest_asyncio.apply()
 
 

diff --git a/tests/unit/examples/test_02-Advanced-NVTabular-workflow.py b/tests/unit/examples/test_02-Advanced-NVTabular-workflow.py
@@ -20,6 +20,7 @@
 from tests.conftest import REPO_ROOT
 
 pytest.importorskip("cudf")
+pytest.importorskip("tensorflow")
 nest_asyncio.apply()