diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 17d6896eda..8304dd2568 100644 --- 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 index 1b788e842d..76c4fb1003 100644 --- 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(