You can just import keras instead of import tensorflow.keras (In my case, Tensorflow version 2.10.1 and Python 3.10.11 and keras version 2.10.0)
import keras
import tensorflow.keras
Example usage in vscode