module 'tensorflow' has no attribute 'contrib'

[relativisticpoint]

I'm trying to run the script hanabi-learning-environment-master/hanabi_learning_environment/agents/rainbow/run_experiment.py but I have an error that says "AttributeError: module 'tensorflow' has no attribute 'contrib'", (I already have tensorflow 2.4.1 and all the necessary modules). I also found this stackoverflow page that addresses this problem, but couldn't upgrade the tensorflow 1.x code to tensorflow 2.x (the same AttributeError occurs when I run train.py because it also imports dqn.py that uses the attribute config of tensorflow which I guess, is no longer available on tensorflow versions 2.X)