Fix tests maybe?

tushuhei · Jul 19, 2016 · 0a108b3 · 0a108b3
1 parent 381a108
commit 0a108b3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/keras/utils/test_utils.py b/keras/utils/test_utils.py
@@ -1,6 +1,7 @@
 import numpy as np
 from numpy.testing import assert_allclose
 import inspect
+import functools
 
 from ..engine import Model, Input
 from ..models import Sequential, model_from_json
@@ -107,6 +108,7 @@ def layer_test(layer_cls, kwargs={}, input_shape=None, input_dtype=None,
 def keras_test(func):
     '''Clean up after tensorflow tests.
     '''
+    @functools.wraps(func)
     def wrapper(*args, **kwargs):
         output = func(*args, **kwargs)
         if K._BACKEND == 'tensorflow':