From 4620c5817d026af801e9d132900a1ae6a5e902de Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Tue, 21 Aug 2018 11:27:34 +0200
Subject: [PATCH] generalise imports in read_activations

---
 keras_visualize_activations/read_activations.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/keras_visualize_activations/read_activations.py b/keras_visualize_activations/read_activations.py
index 053439b..b5d34ab 100644
--- a/keras_visualize_activations/read_activations.py
+++ b/keras_visualize_activations/read_activations.py
@@ -1,7 +1,6 @@
 import keras.backend as K
 
-from keras.engine.input_layer import InputLayer
-from keras.layers.core import Masking
+from keras.layers import Masking, InputLayer
 
 def get_activations(model, model_inputs, print_shape_only=False, layer_name=None):
     print('----- activations -----')
-- 
GitLab