From 0e4e9c6eea2d39b050a56a1e173806e3635e21bc Mon Sep 17 00:00:00 2001 From: dilip patlolla Date: Thu, 19 Dec 2024 11:31:55 -0800 Subject: [PATCH] disable py3.7 tests for mixtral --- tests/benchmarks/model_benchmarks/test_pytorch_mixtral.py | 1 + tests/helper/decorator.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/tests/benchmarks/model_benchmarks/test_pytorch_mixtral.py b/tests/benchmarks/model_benchmarks/test_pytorch_mixtral.py index 7702f7b0e..8c9ad2f35 100644 --- 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 index ff08469ac..b626bb951 100644 --- 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.')