lint fix

microsoft · Nov 20, 2024 · e6f6be3 · e6f6be3
1 parent b28ee17
commit e6f6be3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/benchmarks/model_benchmarks/test_pytorch_llama.py b/tests/benchmarks/model_benchmarks/test_pytorch_llama.py
@@ -50,7 +50,8 @@ def test_pytorch_llama_7b():
     assert (benchmark.return_code == ReturnCode.SUCCESS)
 
     for metric in [
-        'fp8_e4m3_train_step_time', 'fp8_e4m3_train_throughput', 'fp8_e4m3_inference_step_time', 'fp8_e4m3_inference_throughput'
+        'fp8_e4m3_train_step_time', 'fp8_e4m3_train_throughput', 'fp8_e4m3_inference_step_time',
+        'fp8_e4m3_inference_throughput'
     ]:
         assert (len(benchmark.raw_data[metric]) == benchmark.run_count)
         assert (len(benchmark.raw_data[metric][0]) == benchmark._args.num_steps)