From 04cd0aef341221bec17461045ce860cd27e6f174 Mon Sep 17 00:00:00 2001 From: HydrogenSulfate <490868991@qq.com> Date: Sun, 26 Dec 2021 20:34:53 +0800 Subject: [PATCH] change error to IndexError --- .../paddle/fluid/tests/unittests/test_cross_entropy_loss.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py b/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py index adf11e815f..28e286be63 100644 --- a/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py +++ b/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py @@ -1463,7 +1463,7 @@ class TestCrossEntropyFAPIError(unittest.TestCase): weight=weight_data, ignore_index=-100) - self.assertRaises(ValueError, test_WeightLength_NotEqual) + self.assertRaises(IndexError, test_WeightLength_NotEqual) def test_LabelValue_ExceedMax(): input_data = paddle.rand(shape=[20, 100]) @@ -1477,7 +1477,7 @@ class TestCrossEntropyFAPIError(unittest.TestCase): weight=weight_data, ignore_index=-100) - self.assertRaises(ValueError, test_LabelValue_ExceedMax) + self.assertRaises(IndexError, test_LabelValue_ExceedMax) def test_LabelValue_ExceedMin(): input_data = paddle.rand(shape=[20, 100]) -- GitLab