Skip to content

Commit ac0b887

Browse files
committed
fix test
1 parent e9230ff commit ac0b887

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/openvino/test_exporters_cli.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,4 +111,7 @@ def test_exporters_cli_int8(self, task: str, model_type: str):
111111
expected_int8 = _ARCHITECTURES_TO_EXPECTED_INT8[model_type]
112112
for i, model in enumerate(models):
113113
_, num_int8 = get_num_quantized_nodes(model)
114-
self.assertEqual(expected_int8[i], num_int8)
114+
expected = expected_int8[i]
115+
if task == "text-generation":
116+
expected -= 1
117+
self.assertEqual(expected, num_int8)

0 commit comments

Comments
 (0)