forked from mindspore-Ecosystem/mindspore
change ValueError to TypeError for enumerate start
This commit is contained in:
parent
3bb04abc12
commit
60b0ded440
|
@ -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
|
||||
|
||||
|
||||
|
|
|
@ -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…
Reference in New Issue