Unable to get tfmot version from API

Describe the bug
I'm unable to get the tfmot version from the API. I can get the version by using pip list, but I would like to get the version using the API. The version variable is defined in the following file (see link), but somehow it does not work. https://github.com/tensorflow/model-optimization/blob/master/tensorflow_model_optimization/python/core/version.py

System information
TensorFlow version (installed from source or binary): 2.3.0
TensorFlow Model Optimization version (installed from source or binary): 0.5.0
Python version: 3.6.9

Describe the expected behavior
tfmot.__version__ should return the tfmot version instead of an error.

Describe the current behavior
Error message:

Traceback (most recent call last):
  File "test.py", line 8, in <module>
    print('TensorFlow Model Optimization version: '+tfmot.__version__)
AttributeError: module 'tensorflow_model_optimization' has no attribute '__version__'

Code to reproduce the issue

import tensorflow as tf
import tensorflow_model_optimization as tfmot

# Print TensorFlow info
print('TensorFlow version: '+tf.__version__)
print('TensorFlow commit (after -g): '+tf.__git_version__)
print('Keras version: '+tf.keras.__version__)
print('TensorFlow Model Optimization version: '+tfmot.__version__)

Screenshots
If applicable, add screenshots to help explain your problem.

Additional context
Add any other context about the problem here.