O_OFDMNet Query regarding Code

[SreevatsaKolachana]

I am presently using python IDLE 3.10 with tensorflow 2.0.
The utils file is running good but coming to the O-OFDMNet file; there is an error shown which says 'from keras.layers.normalization import BatchNormalization
ModuleNotFoundError: No module named 'keras.layers.normalization'

I am not sure how to correct this since I have installed all the required libraries

Could you help me how can i tackle this?