Fix dependencies

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

Author: beat-buesser

tests/defences/test_neural_cleanse.py

@@ -103,10 +103,10 @@ def test_keras(self):
                 y_test[i, 9] = 1
                 x_test[i, 0:5, 0:5, :] = 1.0
 
-        krc.fit(x_train, y_train, nb_epochs=30)
+        krc.fit(x_train, y_train, nb_epochs=5)
 
         cleanse = NeuralCleanse(krc)
-        defense_cleanse = cleanse(krc, steps=10, patience=10)
+        defense_cleanse = cleanse(krc, steps=3, patience=3)
         defense_cleanse.mitigate(x_test, y_test, mitigation_types=["filtering", "pruning", "unlearning"])
 
         # is_fitted