Tensorflow has no attribute estimator

[Sarah-2021-scu]

Hi,

Thank you for your amazing work. I have been trying to replicate your results but I am encountering the following error:
class RunTrainOpsHook(tf.estimator.SessionRunHook):
^^^^^^^^^^^^
AttributeError: module 'tensorflow' has no attribute 'estimator'
I am including the full stack trace below:
mport run_lib_fastmri
File "/WAVE/datasets/dmlab/sanjum2-data/Subsampled-Brain-MRI-Reconstruction-by-Generative-Adversarial-Neural-Networks/SSIM/score-MRI/run_lib_fastmri.py", line 26, in
import tensorflow_gan as tfgan
File "/WAVE/projects/dmlab/sanjum2-lab/score-mri/lib/python3.12/site-packages/tensorflow_gan/init.py", line 111, in
from tensorflow_gan.python import * # pylint: disable=wildcard-import
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/WAVE/projects/dmlab/sanjum2-lab/score-mri/lib/python3.12/site-packages/tensorflow_gan/python/init.py", line 37, in
from tensorflow_gan.python import estimator
File "/WAVE/projects/dmlab/sanjum2-lab/score-mri/lib/python3.12/site-packages/tensorflow_gan/python/estimator/init.py", line 24, in
from .gan_estimator import *
File "/WAVE/projects/dmlab/sanjum2-lab/score-mri/lib/python3.12/site-packages/tensorflow_gan/python/estimator/gan_estimator.py", line 30, in
from tensorflow_gan.python import train as tfgan_train
File "/WAVE/projects/dmlab/sanjum2-lab/score-mri/lib/python3.12/site-packages/tensorflow_gan/python/train.py", line 1089, in
class RunTrainOpsHook(tf.estimator.SessionRunHook):
^^^^^^^^^^^^
AttributeError: module 'tensorflow' has no attribute 'estimator'

I have tried different combinations of the requirement packages but could not resolve the issue. I really appreciate your help in this regard.