Merge branch 'main' into save-sequence-transform-with-schema-target

NVIDIA-Merlin · Oct 31, 2023 · 86ad222 · 86ad222
2 parents e9dc9fc + 7d99e53
commit 86ad222
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/unit/tf/examples/test_usecase_retrieval_with_hpo.py b/tests/unit/tf/examples/test_usecase_retrieval_with_hpo.py
@@ -3,11 +3,14 @@
 
 from tests.conftest import REPO_ROOT
 
+pytest.importorskip("plotly")
 optuna = pytest.importorskip("optuna")
 
 
 @testbook(
-    REPO_ROOT / "examples/usecases/retrieval-with-hyperparameter-optimization.ipynb", execute=False
+    REPO_ROOT / "examples/usecases/retrieval-with-hyperparameter-optimization.ipynb",
+    execute=False,
+    timeout=120,
 )
 @pytest.mark.notebook
 def test_usecase_pretrained_embeddings(tb):