diff --git a/.circleci/config.yml b/.circleci/config.yml index 65de6541..b1dc2fa0 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -268,7 +268,7 @@ jobs: steps: - checkout - pip_install - - run: pip install torchcsprng hypothesis + - run: pip install torchcsprng hypothesis pytest - unit_tests unittest_py37_torch_release: @@ -277,7 +277,7 @@ jobs: steps: - checkout - pip_install - - run: pip install torchcsprng hypothesis + - run: pip install torchcsprng hypothesis pytest - unit_tests unittest_py38_torch_release: @@ -286,7 +286,7 @@ jobs: steps: - checkout - pip_install - - run: pip install torchcsprng hypothesis + - run: pip install torchcsprng hypothesis pytest - unit_tests unittest_py38_torch_nightly: @@ -304,6 +304,7 @@ jobs: steps: - checkout - pip_install + - run: pip install pytest - mnist_integration_test: device: "cpu" @@ -316,6 +317,7 @@ jobs: - checkout - py_3_6_setup - pip_install + - run: pip install pytest - run_nvidia_smi - mnist_integration_test: device: "cuda" diff --git a/requirements.txt b/requirements.txt index 6d29fc2e..d0dcb5d4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,5 +3,4 @@ torch>=1.3 torchvision>=0.9 tqdm>=4.40 scipy>=1.2 -pytest -requests +requests>=2.25.1 diff --git a/setup.py b/setup.py index f420207c..570dbed0 100644 --- a/setup.py +++ b/setup.py @@ -37,6 +37,7 @@ DEV_REQUIRES = [ "black", + "pytest", "flake8", "sphinx", "sphinx-autodoc-typehints",