Browse Source

删除 test_get_padder_run 中 jittor 的情况

tags/v1.0.0alpha
x54-729 2 years ago
parent
commit
cbffa896c5
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      tests/core/collators/padders/test_get_padder.py

+ 2
- 2
tests/core/collators/padders/test_get_padder.py View File

@@ -14,10 +14,10 @@ def test_get_element_shape_dtype():
catalog = _get_element_shape_dtype([np.zeros(3), np.zeros((2, 1))]) catalog = _get_element_shape_dtype([np.zeros(3), np.zeros((2, 1))])




@pytest.mark.parametrize('backend', ['raw', None, 'numpy', 'torch', 'jittor', 'paddle'])
# @pytest.mark.parametrize('backend', ['raw', None, 'numpy', 'torch', 'jittor', 'paddle'])
@pytest.mark.parametrize('backend', ['raw', None, 'numpy', 'torch', 'paddle'])
@pytest.mark.torch @pytest.mark.torch
@pytest.mark.paddle @pytest.mark.paddle
@pytest.mark.jittor
def test_get_padder_run(backend): def test_get_padder_run(backend):
if not _NEED_IMPORT_TORCH and backend == 'torch': if not _NEED_IMPORT_TORCH and backend == 'torch':
pytest.skip("No torch") pytest.skip("No torch")


Loading…
Cancel
Save