Lint

Author: jackzhxng

extension/llm/export/builder.py

@@ -330,7 +330,9 @@ def pt2e_quantize(self, quantizers: Optional[List[Quantizer]]) -> "LLMEdgeManage
                 assert (
                     self.pre_autograd_graph_module is not None
                 ), "Please run export() first"
-                m = prepare_pt2e(self.pre_autograd_graph_module, composed_quantizer)  # pyre-ignore[6]: In call `prepare_pt2e`, for 1st positional argument, expected `GraphModule` but got `Module`.
+                m = prepare_pt2e(
+                    self.pre_autograd_graph_module, composed_quantizer
+                )  # pyre-ignore[6]: In call `prepare_pt2e`, for 1st positional argument, expected `GraphModule` but got `Module`.
                 logging.info(
                     f"Calibrating with tasks: {self.calibration_tasks}, limit: {self.calibration_limit}, calibration_data: {self.calibration_data}, tokenizer_path: {self.tokenizer_path}, seq_length: {self.calibration_seq_length}"
                 )