Skip to content

Commit

Permalink
Running all tests [run-notebook-tests]
Browse files Browse the repository at this point in the history
  • Loading branch information
ahsimb committed Feb 6, 2024
1 parent a336d25 commit c6f957f
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 7 deletions.
1 change: 0 additions & 1 deletion test/notebooks/nbtest_sagemaker.py
Original file line number Diff line number Diff line change
Expand Up @@ -173,7 +173,6 @@ def continuous_job_polling():
)


@pytest.mark.skip('investigating the notebook test failure')
def test_sagemaker(access_to_temp_secret_store, uploading_hack):

store_path, store_password = access_to_temp_secret_store
Expand Down
12 changes: 6 additions & 6 deletions test/notebooks/nbtest_transformers.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,12 +9,12 @@
"notebook_file",
[
'sequence_classification.ipynb',
pytest.param('question_answering.ipynb', marks=pytest.mark.skip('leaving just one test')),
pytest.param('masked_modelling.ipynb', marks=pytest.mark.skip('leaving just one test')),
pytest.param('token_classification.ipynb', marks=pytest.mark.skip('leaving just one test')),
pytest.param('text_generation.ipynb', marks=pytest.mark.skip('leaving just one test')),
pytest.param('translation.ipynb', marks=pytest.mark.skip('leaving just one test')),
pytest.param('zero_shot_classification.ipynb', marks=pytest.mark.skip('leaving just one test'))
'question_answering.ipynb',
'masked_modelling.ipynb',
'token_classification.ipynb',
'text_generation.ipynb',
'translation.ipynb',
'zero_shot_classification.ipynb'
]
)
def test_transformers(notebook_runner, uploading_hack, notebook_file) -> None:
Expand Down

0 comments on commit c6f957f

Please sign in to comment.