fix mae to mean absolute error

NLeSC · Oct 29, 2019 · 60a8446 · 60a8446
1 parent aba2fe3
commit 60a8446
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_modelgen.py b/tests/test_modelgen.py
@@ -66,7 +66,7 @@ def test_cnn_metrics(self):
         metrics = ['accuracy', 'mae']
         model = modelgen.generate_CNN_model((None, 20, 3), 2, [32, 32], 100, metrics=metrics)
         model_metrics = [m.name for m in model.metrics]
-        assert model_metrics == metrics or model_metrics == ['acc', 'mae']
+        assert model_metrics == metrics or model_metrics == ['acc', 'mean_absolute_error']
 
     def test_CNN_hyperparameters_nrlayers(self):
         """ Number of Conv layers from range [4, 4] should be 4. """