From 173def118f70ffb4dd1e6bf3792e95c2d884731e Mon Sep 17 00:00:00 2001
From: Ludovico Magnocavallo <ludomagno@google.com>
Date: Tue, 21 Dec 2021 11:51:15 +0100
Subject: [PATCH] fix tfdoc replacement (#398)

---
 tools/tfdoc.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/tools/tfdoc.py b/tools/tfdoc.py
index 8b610ceb4..70a50ecac 100755
--- a/tools/tfdoc.py
+++ b/tools/tfdoc.py
@@ -49,7 +49,7 @@ import urllib.parse
 import click
 
 
-__version__ = '2.0'
+__version__ = '2.0.1'
 
 
 # TODO(ludomagno): decide if we want to support variables*.tf and outputs*.tf
@@ -344,7 +344,13 @@ def replace_doc(module_path, doc):
   if doc == result['doc']:
     return
   try:
-    open(readme_path, 'w').write(readme.replace(result['doc'], doc))
+    open(readme_path, 'w').write('\n'.join([
+        readme[:result['start']],
+        MARK_BEGIN,
+        doc,
+        MARK_END,
+        readme[result['end']:]
+    ]))
   except (IOError, OSError) as e:
     raise SystemExit(f'Error replacing README {readme_path}: {e}')