store

xadupre · xadupre · commit 67efd6596748 · 2025-05-05T12:08:29.000+02:00
diff --git a/onnx_diagnostic/torch_export_patches/patch_module.py b/onnx_diagnostic/torch_export_patches/patch_module.py
@@ -264,7 +264,7 @@ def visit_If(self, node):
             )
             if dropped and isinstance(tgt, ast.Tuple):
                 tgt = ast.Tuple(
-                    tuple(t for t in tgt.elts if t.id not in dropped), ctx=ast.Load()
+                    tuple(t for t in tgt.elts if t.id not in dropped), ctx=ast.Store()
                 )
 
             assign = ast.Assign(targets=[tgt], value=call)

Original file line number	Diff line number	Diff line change
`@@ -264,7 +264,7 @@ def visit_If(self, node):`
`264`	`264`	`)`
`265`	`265`	`if dropped and isinstance(tgt, ast.Tuple):`
`266`	`266`	`tgt = ast.Tuple(`
`267`		`- tuple(t for t in tgt.elts if t.id not in dropped), ctx=ast.Load()`
	`267`	`+ tuple(t for t in tgt.elts if t.id not in dropped), ctx=ast.Store()`
`268`	`268`	`)`
`269`	`269`
`270`	`270`	`assign = ast.Assign(targets=[tgt], value=call)`