minor update

microsoft · Oct 26, 2023 · 379903a · 379903a
1 parent 8a995fe
commit 379903a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/test_tools_add_pre_post_processing_to_model.py b/test/test_tools_add_pre_post_processing_to_model.py
@@ -368,6 +368,8 @@ def test_hfbert_tokenizer_optional_output(self):
         s = ort.InferenceSession(str(output_model), so, providers=["CPUExecutionProvider"])
 
         result = s.run(None, {s.get_inputs()[0].name: np.array([[input_text[0], input_text[1]]])})
+
+        self.assertEqual(len(result), 2)
 
         self.assertEqual(np.allclose(result[0], ref_output[0]), True)
         self.assertEqual(np.allclose(result[1], ref_output[1]), True)