Add skipMlFramework

Signed-off-by: Beat Buesser <[email protected]>

Author: Beat Buesser

tests/attacks/evasion/test_shadow_attack.py

@@ -37,6 +37,7 @@ def fix_get_mnist_subset(get_mnist_dataset):
     yield x_train_mnist[:n_train], y_train_mnist[:n_train], x_test_mnist[:n_test], y_test_mnist[:n_test]
 
 
+@pytest.mark.skipMlFramework("keras", "scikitlearn")
 def test_generate(fix_get_mnist_subset, get_image_classifier_list_for_attack):
 
     classifier_list = get_image_classifier_list_for_attack(ShadowAttack)
@@ -61,6 +62,7 @@ def test_generate(fix_get_mnist_subset, get_image_classifier_list_for_attack):
         assert np.max(np.abs(x_train_mnist_adv - x_train_mnist[0:1])) == pytest.approx(0.34966960549354553, 0.06)
 
 
+@pytest.mark.skipMlFramework("keras", "scikitlearn")
 def test_get_regularisation_loss_gradients(fix_get_mnist_subset, get_image_classifier_list_for_attack):
 
     classifier_list = get_image_classifier_list_for_attack(ShadowAttack)