Fixed keras and tf.keras import mix

models/end_to_end.py

@@ -1,12 +1,12 @@
 import math
 
-import keras
+from tensorflow import keras
 import tensorflow as tf
 import numpy as np
 import matplotlib.pyplot as plt
 from matplotlib import collections as matcoll
 from sklearn.preprocessing import OneHotEncoder
-from keras import layers, losses
+from tensorflow.keras import layers, losses
 
 
 class ExtractCentralMessage(layers.Layer):