fix ut

open-mmlab · Aug 28, 2023 · c0af1b9 · c0af1b9
1 parent dd1508d
commit c0af1b9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -62,7 +62,6 @@ jobs:
           python tools/check_env.py
       - name: Run python unittests and generate coverage report
         run: |
-          pytest tests/test_mmcv/test_mmcv_ops.py::test_ONNXNMSop
           coverage run --branch --source mmdeploy -m pytest -rsE tests
           coverage xml
           coverage report -m
@@ -190,7 +189,6 @@ jobs:
       - name: Run unittests and generate coverage report
         id: badge_status
         run: |
-          pytest tests/test_mmcv/test_mmcv_ops.py::test_ONNXNMSop
           coverage run --branch --source mmdeploy -m pytest -rsE tests
           coverage xml
           coverage report -m
@@ -245,7 +243,6 @@ jobs:
           python tools/check_env.py
       - name: Run unittests and generate coverage report
         run: |
-          pytest tests/test_mmcv/test_mmcv_ops.py::test_ONNXNMSop
           coverage run --branch --source mmdeploy -m pytest -rsE tests
           coverage xml
           coverage report -m

diff --git a/mmdeploy/apis/onnx/export.py b/mmdeploy/apis/onnx/export.py
@@ -132,7 +132,7 @@ def wrapper(*arg, **kwargs):
         if isinstance(args, torch.Tensor):
             args = args.cpu()
         elif isinstance(args, (tuple, list)):
-            args = [_.cpu() for _ in args]
+            args = tuple([_.cpu() for _ in args])
         else:
             raise RuntimeError(f'Not supported args: {args}')
         torch.onnx.export(