update test_chicken (#12466)

logits are close, just numerical
2026-01-09 15:08:02 -05:00 · 2025-10-06 15:58:44 +08:00
parent 1b1978b9c0
commit a1881b0c17
1 changed files with 2 additions and 1 deletions
--- a/test/models/test_efficientnet.py
+++ b/test/models/test_efficientnet.py
@@ -101,7 +101,8 @@ class TestResNet(unittest.TestCase):

  def test_chicken(self):
    labels = _infer(self.model, chicken_img)
-    self.assertEqual(_LABELS[labels[0]], "hen")
+    # NOTE: logits for these two are close
+    self.assertIn(_LABELS[labels[0]], ("hen", "cock"))

  def test_car(self):
    labels = _infer(self.model, car_img)