disable py3.7 tests for mixtral

microsoft · Dec 19, 2024 · 0e4e9c6 · 0e4e9c6
1 parent 58c5de9
commit 0e4e9c6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/tests/benchmarks/model_benchmarks/test_pytorch_mixtral.py b/tests/benchmarks/model_benchmarks/test_pytorch_mixtral.py
@@ -10,6 +10,7 @@
 
 @decorator.cuda_test
 @decorator.pytorch_test
+@decorator.python_eol_test
 def test_pytorch_mixtral_8x7b():
     """Test pytorch-mixtral-8x7b benchmark for fp16 train and inference."""
     context = BenchmarkRegistry.create_benchmark_context(

diff --git a/tests/helper/decorator.py b/tests/helper/decorator.py
@@ -4,6 +4,7 @@
 """Unittest decorator helpers."""
 
 import os
+import sys
 import unittest
 import functools
 from pathlib import Path
@@ -12,6 +13,7 @@
 rocm_test = unittest.skipIf(os.environ.get('SB_TEST_ROCM', '0') == '0', 'Skip ROCm tests.')
 
 pytorch_test = unittest.skipIf(os.environ.get('SB_TEST_PYTORCH', '1') == '0', 'Skip PyTorch tests.')
+python_eol_test = unittest.skipIf(sys.version_info < (3, 8), 'Skip tests for Python 3.7 or lower.')
 directx_test = unittest.skipIf(os.environ.get('SB_TEST_DIRECTX', '0') == '0', 'Skip DirectX tests.')