Skip retrieval_with_hpo notebook test when optuna is not installed (#897

) * Skip retrieval_with_hpo notebook test when optuna is not installed * use importorskip instead
NVIDIA-Merlin · Nov 21, 2022 · dc69d42 · dc69d42
1 parent d10938a
commit dc69d42
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/unit/tf/examples/test_usecase_retrieval_with_hpo.py b/tests/unit/tf/examples/test_usecase_retrieval_with_hpo.py
@@ -1,7 +1,10 @@
+import pytest
 from testbook import testbook
 
 from tests.conftest import REPO_ROOT
 
+optuna = pytest.importorskip("optuna")
+
 
 @testbook(
     REPO_ROOT / "examples/usecases/retrieval-with-hyperparameter-optimization.ipynb", execute=False