I have old code using keras instead of tf.keras. I am running on my conda environment with tensorflow 2 and python 3.8 being installed. I installed keras using conda install -c