Fix test assertion in test_trainer.py (#477)

Fix incorrect call to assertTrue to be assertEqual
huggingface · Nov 28, 2023 · d415c9e · d415c9e
1 parent 8c4a1dd
commit d415c9e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_trainer.py b/tests/test_trainer.py
@@ -713,7 +713,7 @@ def test_predict(self):
         trainer = get_regression_trainer(a=1.5, b=2.5, double_output=True, label_names=["labels"])
         preds = trainer.predict(trainer.eval_dataset).predictions
         x = trainer.eval_dataset.x
-        self.assertTrue(len(preds), 2)
+        self.assertEqual(len(preds), 2)
         self.assertTrue(np.allclose(preds[0], 1.5 * x + 2.5))
         self.assertTrue(np.allclose(preds[1], 1.5 * x + 2.5))