Browse Source

!3071 change ValueError to TypeError for enumerate start

Merge pull request !3071 from zhangbuxue/change_ValueError_to_TypeError_for_enumerate_start
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
3e691e54f5
2 changed files with 3 additions and 3 deletions
  1. +2
    -2
      mindspore/_extends/parse/standard_method.py
  2. +1
    -1
      tests/ut/python/pipeline/parse/test_enumerate.py

+ 2
- 2
mindspore/_extends/parse/standard_method.py View File

@@ -135,9 +135,9 @@ def check_is_tuple_or_list(x, op_name, arg_name):
def check_is_const_int(x, op_name, arg_name):
"""check whether x is const int."""
if x is None:
raise ValueError(f"For '{op_name}', the '{arg_name}' should be a const int number, but got not const.")
raise TypeError(f"For '{op_name}', the '{arg_name}' should be a const int number, but got not const.")
if not isinstance(x, int):
raise ValueError(f"For '{op_name}', the '{arg_name}' should be a const int number, but got {x}.")
raise TypeError(f"For '{op_name}', the '{arg_name}' should be a const int number, but got {x}.")
return True




+ 1
- 1
tests/ut/python/pipeline/parse/test_enumerate.py View File

@@ -196,6 +196,6 @@ def test_enumerate_start_type_error():

x = Tensor(np.arange(3 * 4 * 5).reshape((3, 4, 5)))
net = Net()
with pytest.raises(ValueError) as ex:
with pytest.raises(TypeError) as ex:
net((x, x))
assert "For 'enumerate', the 'start'" in str(ex.value)

Loading…
Cancel
Save