From 58ffa2bfb715e6807d0f110b04cfa524922c534d Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Wed, 1 Dec 2021 12:09:23 -0800
Subject: [PATCH] Update pytest for GitHub runner to use --forked with xdist,
 hopefully eliminate memory buildup

---
 .github/workflows/tests.yml | 4 ++--
 tests/test_models.py        | 1 -
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 9e0a4aac..8b49b75d 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -30,7 +30,7 @@ jobs:
     - name: Install testing dependencies
       run: |
         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
       if: startsWith(matrix.os, 'macOS')
       run: pip install --no-cache-dir torch==${{ matrix.torch }} torchvision==${{ matrix.torchvision }}
@@ -48,4 +48,4 @@ jobs:
       env:
         LD_PRELOAD: /usr/lib/x86_64-linux-gnu/libtcmalloc.so.4
       run: |
-        pytest -vv --durations=0 ./tests
+        pytest -vv --forked --durations=0 ./tests
diff --git a/tests/test_models.py b/tests/test_models.py
index 80b1101f..4f80612f 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -350,7 +350,6 @@ if 'GITHUB_ACTIONS' in os.environ:
         'vit_large*',
         'vit_base_patch8*',
         'xcit_large*',
-        '*evob', '*evos', # remove experimental evonorm models, seem to cause issues with dtype manipulation
     ]