Fixed unit test (it checks the first element of the new pair)

cptspacemanspiff · swolchok · commit ad81488323ff · 2025-01-28T13:05:01.000-08:00
diff --git a/exir/tests/test_arg_validator.py b/exir/tests/test_arg_validator.py
@@ -64,7 +64,7 @@ def forward(self, x):
             ops.edge.aten._log_softmax.default.name(),
         )
         self.assertDictEqual(
-            validator.violating_ops[key],
+            validator.violating_ops[key][0],
             {
                 "self": torch.bfloat16,
                 "__ret_0": torch.bfloat16,

Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,7 @@ def forward(self, x):`
`64`	`64`	`ops.edge.aten._log_softmax.default.name(),`
`65`	`65`	`)`
`66`	`66`	`self.assertDictEqual(`
`67`		`- validator.violating_ops[key],`
	`67`	`+ validator.violating_ops[key][0],`
`68`	`68`	`{`
`69`	`69`	`"self": torch.bfloat16,`
`70`	`70`	`"__ret_0": torch.bfloat16,`