fix(test/test_pytorch_ops.py): op_type error

open-mmlab · May 30, 2022 · 9634189 · 9634189
1 parent 5ba1904
commit 9634189
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_pytorch/test_pytorch_ops.py b/tests/test_pytorch/test_pytorch_ops.py
@@ -137,15 +137,15 @@ 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)
         w = torch.rand(2, 3)
         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')