Changes to fix CI

qcdipankar · qcdipankar · commit 95323789dadd · 2025-05-22T07:00:53.000Z
Signed-off-by: Dipankar Sarkar &lt;quic_dipankar@quicinc.com&gt;
diff --git a/tests/peft/test_peft_onnx_transforms.py b/tests/peft/test_peft_onnx_transforms.py
@@ -46,18 +46,19 @@ def test_adapter_weights_to_inputs_transform():
 
     out_onnx, transformed = AdapterWeightsToInputsTransform.apply(test_onnx, adapter_name=adapter_name)
     assert transformed
+
     assert (
         onnx.printer.to_text(out_onnx)
         == textwrap.dedent("""
     <
        ir_version: 8,
        opset_import: ["" : 17]
     >
-    test_adapter_weights (float[n,32] input, float[32,32] layer1.weight, float[32,32] layer2.weight) => (float[n,32] output, float[32,32] layer1.weight_RetainedState, float[32,32] layer2.weight_RetainedState) {
-       layer1output = MatMul (input, layer1.weight)
-       output = MatMul (layer1output, layer2.weight)
-       layer1.weight_RetainedState = Identity (layer1.weight)
-       layer2.weight_RetainedState = Identity (layer2.weight)
+    test_adapter_weights (float[n,32] input, float[32,32] "layer1.weight", float[32,32] "layer2.weight") => (float[n,32] output, float[32,32] "layer1.weight_RetainedState", float[32,32] "layer2.weight_RetainedState") {
+       layer1output = MatMul (input, "layer1.weight")
+       output = MatMul (layer1output, "layer2.weight")
+       ["layer1.weight_identity"] "layer1.weight_RetainedState" = Identity ("layer1.weight")
+       ["layer2.weight_identity"] "layer2.weight_RetainedState" = Identity ("layer2.weight")
     }
     """).strip()
     )

Original file line number	Diff line number	Diff line change
`@@ -46,18 +46,19 @@ def test_adapter_weights_to_inputs_transform():`
`46`	`46`
`47`	`47`	`out_onnx, transformed = AdapterWeightsToInputsTransform.apply(test_onnx, adapter_name=adapter_name)`
`48`	`48`	`assert transformed`
	`49`	`+`
`49`	`50`	`assert (`
`50`	`51`	`onnx.printer.to_text(out_onnx)`
`51`	`52`	`== textwrap.dedent("""`
`52`	`53`	`<`
`53`	`54`	`ir_version: 8,`
`54`	`55`	`opset_import: ["" : 17]`
`55`	`56`	`>`
`56`		`- test_adapter_weights (float[n,32] input, float[32,32] layer1.weight, float[32,32] layer2.weight) => (float[n,32] output, float[32,32] layer1.weight_RetainedState, float[32,32] layer2.weight_RetainedState) {`
`57`		`- layer1output = MatMul (input, layer1.weight)`
`58`		`- output = MatMul (layer1output, layer2.weight)`
`59`		`- layer1.weight_RetainedState = Identity (layer1.weight)`
`60`		`- layer2.weight_RetainedState = Identity (layer2.weight)`
	`57`	`+ test_adapter_weights (float[n,32] input, float[32,32] "layer1.weight", float[32,32] "layer2.weight") => (float[n,32] output, float[32,32] "layer1.weight_RetainedState", float[32,32] "layer2.weight_RetainedState") {`
	`58`	`+ layer1output = MatMul (input, "layer1.weight")`
	`59`	`+ output = MatMul (layer1output, "layer2.weight")`
	`60`	`+ ["layer1.weight_identity"] "layer1.weight_RetainedState" = Identity ("layer1.weight")`
	`61`	`+ ["layer2.weight_identity"] "layer2.weight_RetainedState" = Identity ("layer2.weight")`
`61`	`62`	`}`
`62`	`63`	`""").strip()`
`63`	`64`	`)`