From 9319eff9582bd310ce9b60d3406c1cc72c64c4bf Mon Sep 17 00:00:00 2001 From: zhousiyi Date: Thu, 22 Jul 2021 06:38:12 +0000 Subject: [PATCH] change the error type from RuntimeError to TypeError --- mindspore/ccsrc/frontend/optimizer/ad/kprim.cc | 6 +++--- tests/ut/python/pipeline/infer/test_auto_monad.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/mindspore/ccsrc/frontend/optimizer/ad/kprim.cc b/mindspore/ccsrc/frontend/optimizer/ad/kprim.cc index d9e0f88bb5b..3026e277ac4 100644 --- a/mindspore/ccsrc/frontend/optimizer/ad/kprim.cc +++ b/mindspore/ccsrc/frontend/optimizer/ad/kprim.cc @@ -357,9 +357,9 @@ AnfNodePtr KPrim::BuildOutput(const FuncGraphPtr &bprop_fg, const FuncGraphPtr & } else if (HasAbstractIOMonad(primal_node)) { extra_node = NewValueNode(kIOMonad); } else { - MS_LOG(EXCEPTION) << "Function: " << current_primal_fg->ToString() - << ", has extra parameter which is not UMoand or IOMonad, but: " - << primal_node->DebugString(); + MS_EXCEPTION(TypeError) << "Function: " << current_primal_fg->ToString() + << ", has extra parameter which is not UMoand or IOMonad, but: " + << primal_node->DebugString(); } extra_args.push_back(extra_node); MS_LOG(DEBUG) << "Insert to bprop_fg for node: " << primal_node->DebugString(); diff --git a/tests/ut/python/pipeline/infer/test_auto_monad.py b/tests/ut/python/pipeline/infer/test_auto_monad.py index 1ae95718441..a71f340f36c 100644 --- a/tests/ut/python/pipeline/infer/test_auto_monad.py +++ b/tests/ut/python/pipeline/infer/test_auto_monad.py @@ -181,7 +181,7 @@ def test_user_defined_bad_bprop(): net = GradNet(user_defined_net) x = Tensor(np.array([2.0], np.float32)) y = Tensor(np.array([3.0], np.float32)) - with pytest.raises(RuntimeError): + with pytest.raises(TypeError): net(x, y)