From c76ff4e358307fca98839ee0cd6d3e3eed5f062f Mon Sep 17 00:00:00 2001 From: Radek Osmulski Date: Tue, 24 Oct 2023 08:42:20 +1000 Subject: [PATCH] improve hpo test --- tests/unit/tf/examples/test_usecase_retrieval_with_hpo.py | 6 +++++- 1 file changed, 5 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..64879393d3 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,15 @@ 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):