From ea58e0bc16e9ad97af67eafbb20cf0a43a674ba0 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 27 Jul 2020 18:01:43 -0700 Subject: [PATCH] Disable big models for MacOS test since they are starting to fail frequently. --- tests/test_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_models.py b/tests/test_models.py index 27e9b515..d94bb6b6 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -7,7 +7,7 @@ import fnmatch from timm import list_models, create_model, set_scriptable -if 'GITHUB_ACTIONS' in os.environ and 'Linux' in platform.system(): +if 'GITHUB_ACTIONS' in os.environ: # and 'Linux' in platform.system(): # GitHub Linux runner is slower and hits memory limits sooner than MacOS, exclude bigger models EXCLUDE_FILTERS = ['*efficientnet_l2*', '*resnext101_32x48d'] else: @@ -111,7 +111,7 @@ def test_model_forward_torchscript(model_name, batch_size): EXCLUDE_FEAT_FILTERS = [ '*pruned*', # hopefully fix at some point ] -if 'GITHUB_ACTIONS' in os.environ and 'Linux' in platform.system(): +if 'GITHUB_ACTIONS' in os.environ: # and 'Linux' in platform.system(): # GitHub Linux runner is slower and hits memory limits sooner than MacOS, exclude bigger models EXCLUDE_FEAT_FILTERS += ['*resnext101_32x32d', '*resnext101_32x16d']