diff --git a/setup.py b/setup.py index 282181e33c..3636db6a1f 100644 --- a/setup.py +++ b/setup.py @@ -65,6 +65,7 @@ def readme(): 'flake8', 'pytest', 'pytest-instafail', + 'pytest-timeout', 'pexpect' ], 'docs': [ diff --git a/tests/test_quick_start.py b/tests/test_quick_start.py index bd4f73ef41..96e853e501 100644 --- a/tests/test_quick_start.py +++ b/tests/test_quick_start.py @@ -2,7 +2,6 @@ import json import logging import os -import pickle import shutil import signal import socket @@ -122,15 +121,15 @@ ("classifiers/topics_distilbert_base_uncased.json", "classifiers", ('TI',)): [ONE_ARGUMENT_INFER_CHECK] }, "distil": { - ("classifiers/paraphraser_convers_distilrubert_2L.json", "distil", ('IP')): [TWO_ARGUMENTS_INFER_CHECK], - ("classifiers/paraphraser_convers_distilrubert_6L.json", "distil", ('IP')): [TWO_ARGUMENTS_INFER_CHECK], - ("classifiers/rusentiment_convers_distilrubert_2L.json", "distil", ('IP')): [ONE_ARGUMENT_INFER_CHECK], - ("classifiers/rusentiment_convers_distilrubert_6L.json", "distil", ('IP')): [ONE_ARGUMENT_INFER_CHECK], - ("ner/ner_rus_convers_distilrubert_2L.json", "distil", ('IP')): [ONE_ARGUMENT_INFER_CHECK], - ("ner/ner_rus_convers_distilrubert_6L.json", "distil", ('IP')): [ONE_ARGUMENT_INFER_CHECK], - ("ner/ner_case_agnostic_mdistilbert.json", "distil", ('IP')): [ONE_ARGUMENT_INFER_CHECK], - ("squad/squad_ru_convers_distilrubert_2L.json", "distil", ('IP')): [TWO_ARGUMENTS_INFER_CHECK], - ("squad/squad_ru_convers_distilrubert_6L.json", "distil", ('IP')): [TWO_ARGUMENTS_INFER_CHECK] + ("classifiers/paraphraser_convers_distilrubert_2L.json", "distil", ('IP',)): [TWO_ARGUMENTS_INFER_CHECK], + ("classifiers/paraphraser_convers_distilrubert_6L.json", "distil", ('IP',)): [TWO_ARGUMENTS_INFER_CHECK], + ("classifiers/rusentiment_convers_distilrubert_2L.json", "distil", ('IP',)): [ONE_ARGUMENT_INFER_CHECK], + ("classifiers/rusentiment_convers_distilrubert_6L.json", "distil", ('IP',)): [ONE_ARGUMENT_INFER_CHECK], + ("ner/ner_rus_convers_distilrubert_2L.json", "distil", ('IP',)): [ONE_ARGUMENT_INFER_CHECK], + ("ner/ner_rus_convers_distilrubert_6L.json", "distil", ('IP',)): [ONE_ARGUMENT_INFER_CHECK], + ("ner/ner_case_agnostic_mdistilbert.json", "distil", ('IP',)): [ONE_ARGUMENT_INFER_CHECK], + ("squad/squad_ru_convers_distilrubert_2L.json", "distil", ('IP',)): [TWO_ARGUMENTS_INFER_CHECK], + ("squad/squad_ru_convers_distilrubert_6L.json", "distil", ('IP',)): [TWO_ARGUMENTS_INFER_CHECK] }, "russian_super_glue": { ("russian_super_glue/russian_superglue_lidirus_rubert.json", "russian_super_glue", ('IP',)): [TWO_ARGUMENTS_INFER_CHECK], diff --git a/utils/Docker/cmd.sh b/utils/Docker/cmd.sh index 1529d8f524..321b67745d 100755 --- a/utils/Docker/cmd.sh +++ b/utils/Docker/cmd.sh @@ -13,4 +13,4 @@ cd .. flake8 `python -c 'import deeppavlov; print(deeppavlov.__path__[0])'` --count --select=E9,F63,F7,F82 --show-source --statistics -pytest -v --disable-warnings --instafail $PYTEST_ARGS +pytest -v --disable-warnings --instafail --timeout=1200 $PYTEST_ARGS