diff --git a/test/registered/amd/accuracy/mi35x/test_deepseek_r1_mxfp4_tp2_mi35x.py b/test/registered/amd/accuracy/mi35x/test_deepseek_r1_mxfp4_tp2_mi35x.py index 12ed608d8..36cfc9a00 100644 --- a/test/registered/amd/accuracy/mi35x/test_deepseek_r1_mxfp4_tp2_mi35x.py +++ b/test/registered/amd/accuracy/mi35x/test_deepseek_r1_mxfp4_tp2_mi35x.py @@ -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")) diff --git a/test/registered/amd/accuracy/mi35x/test_deepseek_r1_mxfp4_tp4_mi35x.py b/test/registered/amd/accuracy/mi35x/test_deepseek_r1_mxfp4_tp4_mi35x.py index a6d977c53..f484f46f1 100644 --- a/test/registered/amd/accuracy/mi35x/test_deepseek_r1_mxfp4_tp4_mi35x.py +++ b/test/registered/amd/accuracy/mi35x/test_deepseek_r1_mxfp4_tp4_mi35x.py @@ -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"))