update format2

Author: YijiaZhao

collector/collect.py

@@ -375,7 +375,7 @@ def collect_trtllm(num_processes: int, ops: list[str] | None = None):
             "module": "trtllm.collect_mla",
             "get_func": "get_context_mla_test_cases",
             "run_func": "run_mla",
-            "version_handler": lambda v: "trtllm.collect_mla_1_1rc2" 
+            "version_handler": lambda v: "trtllm.collect_mla_1_1rc2"
             if v.startswith(("1.1.0", "1.2.0")) else "trtllm.collect_mla",
         },
         {
@@ -384,7 +384,7 @@ def collect_trtllm(num_processes: int, ops: list[str] | None = None):
             "module": "trtllm.collect_mla",
             "get_func": "get_generation_mla_test_cases",
             "run_func": "run_mla",
-            "version_handler": lambda v: "trtllm.collect_mla_1_1rc2" 
+            "version_handler": lambda v: "trtllm.collect_mla_1_1rc2"
             if v.startswith(("1.1.0", "1.2.0")) else "trtllm.collect_mla",
         },
         # Attention collections - separate entries for context and generation

collector/trtllm/collect_mla_1_1rc2.py

@@ -25,6 +25,7 @@
 
 from helper import log_perf
 
+
 def get_context_mla_test_cases():
     dtype_list = [tensorrt_llm.bindings.DataType.BF16, tensorrt_llm.bindings.DataType.FP8]
     test_cases = []