From e4be207451a1f2a42393928d9d5aa81355b271ad Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=A4=A9=E9=82=91?= Date: Wed, 9 Oct 2024 10:42:38 +0800 Subject: [PATCH] fix sampler test --- tzrec/datasets/sampler_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tzrec/datasets/sampler_test.py b/tzrec/datasets/sampler_test.py index 311937e..c7b861f 100644 --- a/tzrec/datasets/sampler_test.py +++ b/tzrec/datasets/sampler_test.py @@ -457,9 +457,9 @@ def _sampler_worker(pos_res, neg_res): p.join() if p.exitcode != 0: raise RuntimeError("worker failed.") - self.assertEqual(len(pos_res["int_a"]), 4 * 5) - self.assertEqual(len(pos_res["float_b"]), 4 * 5) - self.assertEqual(len(pos_res["str_c"]), 4 * 5) + self.assertEqual(len(pos_res["int_a"]), 4 * 4) + self.assertEqual(len(pos_res["float_b"]), 4 * 4) + self.assertEqual(len(pos_res["str_c"]), 4 * 4) self.assertEqual(len(neg_res["int_a"]), 4 * 15) self.assertEqual(len(neg_res["float_b"]), 4 * 15) self.assertEqual(len(neg_res["str_c"]), 4 * 15)