From 96341896f00e6a8a6dc447189a71d5e599ff8c8d Mon Sep 17 00:00:00 2001 From: tpoisonooo Date: Mon, 30 May 2022 11:02:02 +0800 Subject: [PATCH] fix(test/test_pytorch_ops.py): op_type error --- tests/test_pytorch/test_pytorch_ops.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_pytorch/test_pytorch_ops.py b/tests/test_pytorch/test_pytorch_ops.py index 5cefefa4fe..f8963bc5ad 100644 --- a/tests/test_pytorch/test_pytorch_ops.py +++ b/tests/test_pytorch/test_pytorch_ops.py @@ -137,7 +137,7 @@ def test_normal(self): model = OpModel(torch.nn.functional.linear, w, bias).eval() nodes = get_model_onnx_nodes(model, x) print(nodes) - assert nodes[0].op_type == 'Matmal' + assert nodes[0].op_type == 'MatMul' def test_no_bias(self): x = torch.rand(1, 2, 3) @@ -145,7 +145,7 @@ def test_no_bias(self): model = OpModel(torch.nn.functional.linear, w).eval() nodes = get_model_onnx_nodes(model, x) print(nodes) - assert nodes[0].op_type == 'Matmal' + assert nodes[0].op_type == 'MatMul' @pytest.mark.usefixtures('prepare_symbolics')