Reformated onnx_transform file

Signed-off-by: abhishek-singh591 <[email protected]>

Author: abhishek-singh591

QEfficient/base/onnx_transforms.py

@@ -6,11 +6,11 @@
 # ----------------------------------------------------------------------------
 
 import os
-import numpy as np
-from typing import Optional, Tuple
 from concurrent.futures import ThreadPoolExecutor
-from onnx import ModelProto, external_data_helper, numpy_helper
+from typing import Optional, Tuple
 
+import numpy as np
+from onnx import ModelProto, external_data_helper, numpy_helper
 
 
 class OnnxTransform:
@@ -91,4 +91,4 @@ def process_tensor(args):
 
         with ThreadPoolExecutor(max_workers=os.cpu_count() * 4) as executor:
             list(executor.map(process_tensor, file_assignments))
-        return model, transformed
+        return model, transformed