From 7d99e53e6b8521cfe85c0ab7bbe8ea150289ecf4 Mon Sep 17 00:00:00 2001
From: Radek Osmulski <rosmulski@gmail.com>
Date: Tue, 31 Oct 2023 13:25:41 +1000
Subject: [PATCH] improve the hpo test (#1222)

* improve the hpo test
---
 tests/unit/tf/examples/test_usecase_retrieval_with_hpo.py | 5 ++++-
 1 file changed, 4 insertions(+), 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
index 4fabbd8d14..f1727ba27d 100644
--- 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):