From fe785b5be41da36743bc9af20a68a1e8857ed8fc Mon Sep 17 00:00:00 2001 From: miguelgfierro Date: Fri, 26 Jul 2019 14:41:48 +0100 Subject: [PATCH] :bug: --- tests/integration/test_notebooks_sentence_similarity.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/integration/test_notebooks_sentence_similarity.py b/tests/integration/test_notebooks_sentence_similarity.py index e438d19..732b31f 100644 --- a/tests/integration/test_notebooks_sentence_similarity.py +++ b/tests/integration/test_notebooks_sentence_similarity.py @@ -5,7 +5,7 @@ import sys import pytest import papermill as pm import scrapbook as sb -from tests.notebooks_common import OUTPUT_NOTEBOOK +from tests.notebooks_common import OUTPUT_NOTEBOOK, KERNEL_NAME sys.path.append("../../") ABS_TOL = 0.2 @@ -37,7 +37,7 @@ def baseline_results(): @pytest.mark.integration def test_similarity_embeddings_baseline_runs(notebooks, baseline_results): notebook_path = notebooks["similarity_embeddings_baseline"] - pm.execute_notebook(notebook_path, OUTPUT_NOTEBOOK) + pm.execute_notebook(notebook_path, OUTPUT_NOTEBOOK, kernel_name=KERNEL_NAME) results = sb.read_notebook(OUTPUT_NOTEBOOK).scraps.data_dict["results"] for key, value in baseline_results.items(): assert results[key] == pytest.approx(value, abs=ABS_TOL) @@ -50,6 +50,7 @@ def test_gensen_local(notebooks): pm.execute_notebook( notebook_path, OUTPUT_NOTEBOOK, + kernel_name=KERNEL_NAME, parameters=dict( max_epoch=1, config_filepath="../../scenarios/sentence_similarity/gensen_config.json",