Skip to content

Commit 0e4e9c6

Browse files
committed
disable py3.7 tests for mixtral
1 parent 58c5de9 commit 0e4e9c6

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

tests/benchmarks/model_benchmarks/test_pytorch_mixtral.py

+1
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010

1111
@decorator.cuda_test
1212
@decorator.pytorch_test
13+
@decorator.python_eol_test
1314
def test_pytorch_mixtral_8x7b():
1415
"""Test pytorch-mixtral-8x7b benchmark for fp16 train and inference."""
1516
context = BenchmarkRegistry.create_benchmark_context(

tests/helper/decorator.py

+2
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
"""Unittest decorator helpers."""
55

66
import os
7+
import sys
78
import unittest
89
import functools
910
from pathlib import Path
@@ -12,6 +13,7 @@
1213
rocm_test = unittest.skipIf(os.environ.get('SB_TEST_ROCM', '0') == '0', 'Skip ROCm tests.')
1314

1415
pytorch_test = unittest.skipIf(os.environ.get('SB_TEST_PYTORCH', '1') == '0', 'Skip PyTorch tests.')
16+
python_eol_test = unittest.skipIf(sys.version_info < (3, 8), 'Skip tests for Python 3.7 or lower.')
1517
directx_test = unittest.skipIf(os.environ.get('SB_TEST_DIRECTX', '0') == '0', 'Skip DirectX tests.')
1618

1719

0 commit comments

Comments
 (0)