fixing travis build

keras-team · Feb 18, 2019 · 45cd51f · 45cd51f
1 parent fcab9cb
commit 45cd51f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/keras_contrib/wrappers/cdropout.py b/keras_contrib/wrappers/cdropout.py
@@ -1,11 +1,12 @@
 # -*- coding: utf-8 -*-
+from __future__ import absolute_import
 
 import numpy as np
 
 from keras import backend as K
 from keras.engine import InputSpec
 from keras.initializers import RandomUniform
-from keras.layers import Wrapper
+from keras.layers.wrappers import Wrapper
 
 
 class ConcreteDropout(Wrapper):