adding import or skip for tensorflow framework required by examples tests (#1691)

jperez999 · web-flow · commit 8f87c3471604 · 2022-10-07T20:19:49.000-04:00
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()
 
 

Original file line number	Diff line number	Diff line change
`@@ -14,10 +14,12 @@`
`14`	`14`	`# limitations under the License.`
`15`	`15`	`#`
`16`	`16`	`import nest_asyncio`
	`17`	`+import pytest`
`17`	`18`	`from testbook import testbook`
`18`	`19`
`19`	`20`	`from tests.conftest import REPO_ROOT`
`20`	`21`
	`22`	`+pytest.importorskip("tensorflow")`
`21`	`23`	`nest_asyncio.apply()`
`22`	`24`
`23`	`25`