fixed broken keras import

models/end_to_end.py

@@ -1,11 +1,10 @@
 import math
 
-import keras
 import tensorflow as tf
 import numpy as np
 import matplotlib.pyplot as plt
 from sklearn.preprocessing import OneHotEncoder
-from keras import layers, losses
+from tensorflow.keras import layers, losses
 
 
 class ExtractCentralMessage(layers.Layer):
@@ -230,7 +229,7 @@ class EndToEndAutoencoder(tf.keras.Model):
         X_train, y_train = self.generate_random_inputs(int(num_of_blocks*train_size))
         X_test, y_test = self.generate_random_inputs(int(num_of_blocks*(1-train_size)))
 
-        opt = keras.optimizers.Adam(learning_rate=lr)
+        opt = tf.keras.optimizers.Adam(learning_rate=lr)
 
         self.compile(optimizer=opt,
                      loss=losses.BinaryCrossentropy(),