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 d8f541bf48c31c9bba0151a33f305d9272990210..adf11e815faa97bb547b9b1d50f839baa56e9595 100644 --- a/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py +++ b/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py @@ -1477,7 +1477,7 @@ class TestCrossEntropyFAPIError(unittest.TestCase): weight=weight_data, ignore_index=-100) - self.assertRaises(IndexError, test_LabelValue_ExceedMax) + self.assertRaises(ValueError, test_LabelValue_ExceedMax) def test_LabelValue_ExceedMin(): input_data = paddle.rand(shape=[20, 100]) @@ -1491,7 +1491,7 @@ class TestCrossEntropyFAPIError(unittest.TestCase): weight=weight_data, ignore_index=-100) - self.assertRaises(IndexError, test_LabelValue_ExceedMin) + self.assertRaises(ValueError, test_LabelValue_ExceedMin) def static_test_WeightLength_NotEqual(): input_np = np.random.random([2, 4]).astype('float32')