Make test_mnist runnable by pytest and directly

This commit is contained in:
Adrian Garcia Badaracco
2020-10-21 11:30:08 -05:00
parent 5afe6b1f68
commit 02adb0ac3a
2 changed files with 54 additions and 50 deletions

View File

@@ -17,9 +17,6 @@ jobs:
with:
python-version: 3.8
- name: Install Dependencies
run: pip install ipython numpy tqdm requests torch
- name: Run mnist test
run: ipython3 test/mnist.py
- name: Run compare to torch test
run: ipython3 test/test.py
run: pip install pytest numpy tqdm requests torch
- name: Run Pytest
run: python -m pytest -s -v