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']