# 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