Handle keras and imbalanced_learn dependencies

[liellnima]

add something to rewrite import in keras_self_attention:
from tensorflow import keras
instead of: import keras

TODO add: keras_requires=">=2, <3",
TODO add: imbalanced_learn >= 0.12.0

[liellnima]

unclear how big of an issue - could be a quick fix or could be tideous