|
|
@@ -17,6 +17,7 @@ def test_get_element_shape_dtype(): |
|
|
|
@pytest.mark.parametrize('backend', ['raw', None, 'numpy', 'torch', 'jittor', 'paddle']) |
|
|
|
@pytest.mark.torch |
|
|
|
@pytest.mark.paddle |
|
|
|
@pytest.mark.jittor |
|
|
|
def test_get_padder_run(backend): |
|
|
|
if not _NEED_IMPORT_TORCH and backend == 'torch': |
|
|
|
pytest.skip("No torch") |
|
|
|