change ValueError to TypeError for enumerate start

This commit is contained in:
buxue 2020-07-15 10:15:52 +08:00
parent 3bb04abc12
commit 60b0ded440
2 changed files with 3 additions and 3 deletions

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

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)