File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change 1414from executorch .backends .arm .test .tester .arm_tester import ArmTester
1515from parameterized import parameterized
1616
17- class OpCheckError (Exception ):
18- pass
1917
2018def test_rescale_op ():
2119 sample_inputs = [
@@ -65,7 +63,7 @@ def test_nonzero_zp_for_int32():
6563 ),
6664 ]
6765 for sample_input in sample_inputs :
68- with pytest .raises (OpCheckError ):
66+ with pytest .raises (Exception , match = " OpCheckError" ):
6967 torch .library .opcheck (torch .ops .tosa ._rescale , sample_input )
7068
7169
@@ -88,7 +86,7 @@ def test_zp_outside_range():
8886 ),
8987 ]
9088 for sample_input in sample_inputs :
91- with pytest .raises (OpCheckError ):
89+ with pytest .raises (Exception , match = " OpCheckError" ):
9290 torch .library .opcheck (torch .ops .tosa ._rescale , sample_input )
9391
9492
You can’t perform that action at this time.
0 commit comments