Browse Source

change error type

tags/v0.3.0-alpha
zhaojichen 6 years ago
parent
commit
d1add07b5a
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      tests/ut/python/pynative_mode/nn/test_pooling.py

+ 1
- 1
tests/ut/python/pynative_mode/nn/test_pooling.py View File

@@ -38,7 +38,7 @@ def test_avgpool2d_error_input():
""" test_avgpool2d_error_input """
kernel_size = 5
stride = 2.3
with pytest.raises(ValueError):
with pytest.raises(TypeError):
nn.AvgPool2d(kernel_size, stride)




Loading…
Cancel
Save