Quality fixes

neuralmagic · Oct 20, 2023 · f3cbf3d · f3cbf3d
1 parent f1de171
commit f3cbf3d
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/tests/deepsparse/transformers/test_helpers.py b/tests/deepsparse/transformers/test_helpers.py
@@ -183,11 +183,7 @@ def test_truncate_transformer_onnx_model(
     model_onnx_path = get_model_onnx_path(model_name)
     output_name = "embedding"
 
-    (
-        truncated_onnx_path,
-        output_names,
-        _,
-    ) = truncate_transformer_onnx_model(
+    (truncated_onnx_path, output_names, _,) = truncate_transformer_onnx_model(
         model_path=model_onnx_path,
         emb_extraction_layer=emb_extraction_layer,
         hidden_layer_size=None,