Update pytest for GitHub runner to use --forked with xdist, hopefully eliminate memory buildup

pull/1014/head
Ross Wightman 3 years ago
parent 78912b6375
commit 58ffa2bfb7

@ -30,7 +30,7 @@ jobs:
- name: Install testing dependencies - name: Install testing dependencies
run: | run: |
python -m pip install --upgrade pip python -m pip install --upgrade pip
pip install pytest pytest-timeout expecttest pip install pytest pytest-timeout pytest-xdist expecttest
- name: Install torch on mac - name: Install torch on mac
if: startsWith(matrix.os, 'macOS') if: startsWith(matrix.os, 'macOS')
run: pip install --no-cache-dir torch==${{ matrix.torch }} torchvision==${{ matrix.torchvision }} run: pip install --no-cache-dir torch==${{ matrix.torch }} torchvision==${{ matrix.torchvision }}
@ -48,4 +48,4 @@ jobs:
env: env:
LD_PRELOAD: /usr/lib/x86_64-linux-gnu/libtcmalloc.so.4 LD_PRELOAD: /usr/lib/x86_64-linux-gnu/libtcmalloc.so.4
run: | run: |
pytest -vv --durations=0 ./tests pytest -vv --forked --durations=0 ./tests

@ -350,7 +350,6 @@ if 'GITHUB_ACTIONS' in os.environ:
'vit_large*', 'vit_large*',
'vit_base_patch8*', 'vit_base_patch8*',
'xcit_large*', 'xcit_large*',
'*evob', '*evos', # remove experimental evonorm models, seem to cause issues with dtype manipulation
] ]

Loading…
Cancel
Save