Fix lint break on main (#28886)

This commit is contained in:
Liangsheng Yin
2026-06-21 22:46:08 -07:00
committed by GitHub
parent e2540188ce
commit 64e455d4bf
2 changed files with 6 additions and 2 deletions
@@ -112,7 +112,9 @@ def run_gsm8k_benchmark(
class TestDeepSeekR1MXFP4TP2MI35x(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.model = os.environ.get("DEEPSEEK_R1_MXFP4_MODEL_PATH", "amd/DeepSeek-R1-MXFP4-Preview")
cls.model = os.environ.get(
"DEEPSEEK_R1_MXFP4_MODEL_PATH", "amd/DeepSeek-R1-MXFP4-Preview"
)
cls.base_url = DEFAULT_URL_FOR_TEST
cls.num_questions = int(os.environ.get("GSM8K_NUM_QUESTIONS", "1319"))
@@ -111,7 +111,9 @@ def run_gsm8k_benchmark(
class TestDeepSeekR1MXFP4TP4MI35x(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.model = os.environ.get("DEEPSEEK_R1_MXFP4_MODEL_PATH", "amd/DeepSeek-R1-MXFP4-Preview")
cls.model = os.environ.get(
"DEEPSEEK_R1_MXFP4_MODEL_PATH", "amd/DeepSeek-R1-MXFP4-Preview"
)
cls.base_url = DEFAULT_URL_FOR_TEST
cls.num_questions = int(os.environ.get("GSM8K_NUM_QUESTIONS", "1319"))