Skip to content

Commit b65f173

Browse files
committed
Run other methods with PyTorch as well
Remove PyTorch from the skipMlFramework argument. Signed-off-by: Hessel Tuinhof <[email protected]>
1 parent 6550de9 commit b65f173

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tests/attacks/evasion/test_imperceptible_asr.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -34,21 +34,21 @@ class TestImperceptibleASR:
3434
Test the ImperceptibleASR attack.
3535
"""
3636

37-
@pytest.mark.skipMlFramework("pytorch", "tensorflow1", "tensorflow2", "mxnet", "kerastf", "non_dl_frameworks")
37+
@pytest.mark.skipMlFramework("tensorflow1", "tensorflow2", "mxnet", "kerastf", "non_dl_frameworks")
3838
def test_is_subclass(self, art_warning):
3939
try:
4040
assert issubclass(ImperceptibleASR, EvasionAttack)
4141
except ARTTestException as e:
4242
art_warning(e)
4343

44-
@pytest.mark.skipMlFramework("pytorch", "tensorflow1", "tensorflow2", "mxnet", "kerastf", "non_dl_frameworks")
44+
@pytest.mark.skipMlFramework("tensorflow1", "tensorflow2", "mxnet", "kerastf", "non_dl_frameworks")
4545
def test_implements_abstract_methods(self, art_warning, asr_dummy_estimator):
4646
try:
4747
ImperceptibleASR(estimator=asr_dummy_estimator(), masker=PsychoacousticMasker())
4848
except ARTTestException as e:
4949
art_warning(e)
5050

51-
@pytest.mark.skipMlFramework("pytorch", "tensorflow1", "tensorflow2", "mxnet", "kerastf", "non_dl_frameworks")
51+
@pytest.mark.skipMlFramework("tensorflow1", "tensorflow2", "mxnet", "kerastf", "non_dl_frameworks")
5252
def test_generate(self, art_warning, mocker, asr_dummy_estimator, audio_data):
5353
try:
5454
test_input, test_target = audio_data
@@ -64,7 +64,7 @@ def test_generate(self, art_warning, mocker, asr_dummy_estimator, audio_data):
6464
except ARTTestException as e:
6565
art_warning(e)
6666

67-
@pytest.mark.skipMlFramework("pytorch", "tensorflow1", "tensorflow2", "mxnet", "kerastf", "non_dl_frameworks")
67+
@pytest.mark.skipMlFramework("tensorflow1", "tensorflow2", "mxnet", "kerastf", "non_dl_frameworks")
6868
def test_generate_batch(self, art_warning, mocker, asr_dummy_estimator, audio_data):
6969
try:
7070
test_input, test_target = audio_data
@@ -80,7 +80,7 @@ def test_generate_batch(self, art_warning, mocker, asr_dummy_estimator, audio_da
8080
except ARTTestException as e:
8181
art_warning(e)
8282

83-
@pytest.mark.skipMlFramework("pytorch", "tensorflow1", "tensorflow2", "mxnet", "kerastf", "non_dl_frameworks")
83+
@pytest.mark.skipMlFramework("tensorflow1", "tensorflow2", "mxnet", "kerastf", "non_dl_frameworks")
8484
def test_create_adversarial(self, art_warning, mocker, asr_dummy_estimator, audio_data):
8585
try:
8686
test_input, test_target = audio_data
@@ -107,7 +107,7 @@ def test_create_adversarial(self, art_warning, mocker, asr_dummy_estimator, audi
107107
except ARTTestException as e:
108108
art_warning(e)
109109

110-
@pytest.mark.skipMlFramework("pytorch", "tensorflow1", "tensorflow2", "mxnet", "kerastf", "non_dl_frameworks")
110+
@pytest.mark.skipMlFramework("tensorflow1", "tensorflow2", "mxnet", "kerastf", "non_dl_frameworks")
111111
def test_create_imperceptible(self, art_warning, mocker, asr_dummy_estimator, audio_data):
112112
try:
113113
test_input, test_target = audio_data

0 commit comments

Comments
 (0)