Not able to import quantize_model and quantize_aware_training

Working on quantization aware training but not able to import quantize_model and quantize_aware_training
Below are the imports I am using
from tensorflow_model_optimization.python.core.quantization.keras import quantize_aware_training
from tensorflow_model_optimization.python.core.quantization.keras.quantize_aware_training import quantize_model
Getting below error:
ImportError: cannot import name ‘quantize_aware_training’ from ‘tensorflow_model_optimization.python.core.quantization.keras’ (/usr/local/lib/python3.10/dist-packages/tensorflow_model_optimization/python/core/quantization/keras/init.py)
Installed packages are up to date versions and compatible with tensorflow and tensorflow-model-optimization. What is the way to resolve the issue?