I have imported Keras from tensor flow, as so :
import numpy as np import tensorflow as tf from tensorflow.keras import layers print("working")