From 8f87c3471604d44937ea97363841b8faee70872c Mon Sep 17 00:00:00 2001
From: Julio Perez <37191411+jperez999@users.noreply.github.com>
Date: Fri, 7 Oct 2022 20:19:49 -0400
Subject: [PATCH] adding import or skip for tensorflow framework required by
 examples tests (#1691)

---
 tests/unit/examples/test_01-Getting-started.py             | 2 ++
 tests/unit/examples/test_02-Advanced-NVTabular-workflow.py | 1 +
 2 files changed, 3 insertions(+)

diff --git a/tests/unit/examples/test_01-Getting-started.py b/tests/unit/examples/test_01-Getting-started.py
index 99d51ec09bf..1ddfbe67c6f 100644
--- 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
index 83a88a7372c..3bc54af5a89 100644
--- 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()