[MNT] Removing tensorflow_addons
dependency
#1421
Merged
tensorflow_addons
dependency
#1421