Ejecutando el estimador de tensorflow hoy y se me ocurrió este error, ¿alguna idea de cómo resolverlo?

 File "C:\Users\ASUS Laptop\anaconda3\envs\tf_gpu\lib\site-packages\tensorflow_estimator\python\estimator\estimator.py", line 70, in <module> @doc_controls.inheritable_header("""\ AttributeError: module 'tensorflow.tools.docs.doc_controls' has no attribute 'inheritable_header'

Esto está en mi archivo estimator.py: @doc_controls.inheritable_header("""
Advertencia: No se recomiendan estimadores para código nuevo. Los estimadores ejecutan código de estilo v1.Session , que es más difícil de escribir correctamente y puede comportarse de manera inesperada, especialmente cuando se combina con el código TF 2. Los estimadores están incluidos en nuestras garantías de compatibilidad , pero no recibirán correcciones aparte de las vulnerabilidades de seguridad. Consulte la guía de migración para obtener más información. """)