I'm using [email protected] and [email protected] It seems like the issue can be solved by simply changing
from tensorflow.keras.layers import Dense # for example
to
from keras.layers import Dense