Quality fixes

anmarques · anmarques · commit 8d63a8a38888 · 2023-10-03T13:05:42.000-04:00
diff --git a/src/sparseml/exporters/onnx_to_deepsparse.py b/src/sparseml/exporters/onnx_to_deepsparse.py
@@ -80,7 +80,9 @@ def __init__(
             sparseml_transforms.PropagateDequantThroughSplit(),
         ]
         if use_qlinear_matmul:
-            transforms.append(sparseml_transforms.MatMulToQLinearMatMul(),)
+            transforms.append(
+                sparseml_transforms.MatMulToQLinearMatMul(),
+            )
 
         transforms.extend(
             [

Original file line number	Diff line number	Diff line change
`@@ -80,7 +80,9 @@ def __init__(`
`80`	`80`	`sparseml_transforms.PropagateDequantThroughSplit(),`
`81`	`81`	`]`
`82`	`82`	`if use_qlinear_matmul:`
`83`		`- transforms.append(sparseml_transforms.MatMulToQLinearMatMul(),)`
	`83`	`+ transforms.append(`
	`84`	`+ sparseml_transforms.MatMulToQLinearMatMul(),`
	`85`	`+ )`
`84`	`86`
`85`	`87`	`transforms.extend(`
`86`	`88`	`[`