37 строки
1.3 KiB
Python
37 строки
1.3 KiB
Python
# Copyright (c) Microsoft Corporation. All rights reserved.
|
||
# Licensed under the MIT License.
|
||
|
||
# NOTE: This file is used by pytest to inject fixtures automatically. As it is explained in the documentation
|
||
# https://docs.pytest.org/en/latest/fixture.html:
|
||
# "If during implementing your tests you realize that you want to use a fixture function from multiple test files
|
||
# you can move it to a conftest.py file. You don’t need to import the fixture you want to use in a test, it
|
||
# automatically gets discovered by pytest."
|
||
|
||
import pytest
|
||
import os
|
||
from tests.notebooks_common import path_notebooks
|
||
|
||
|
||
@pytest.fixture(scope="module")
|
||
def notebooks():
|
||
folder_notebooks = path_notebooks()
|
||
|
||
# Path for the notebooks
|
||
paths = {
|
||
"msrpc": os.path.join(
|
||
folder_notebooks,"data_prep", "msrpc.ipynb"
|
||
),
|
||
"snli": os.path.join(
|
||
folder_notebooks,"data_prep", "snli.ipynb"
|
||
),
|
||
"stsbenchmark": os.path.join(
|
||
folder_notebooks,"data_prep", "stsbenchmark.ipynb"
|
||
),
|
||
"similarity_embeddings_baseline": os.path.join(
|
||
folder_notebooks,"sentence_similarity", "baseline_deep_dive.ipynb"
|
||
),
|
||
"embedding_trainer": os.path.join(
|
||
folder_notebooks,"embeddings", "embedding_trainer.ipynb"
|
||
),
|
||
}
|
||
return paths |