diff --git a/docs/requirements.txt b/docs/requirements.txt
index 8a1cee6e..616a0b73 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -6,4 +6,4 @@ sphinx-rtd-theme
 
 lightning >= 2.0.0, < 3.0.0a0
 pytorch-lightning >= 1.5.0, < 2.0.0a0
-tensorflow-cpu >= 2.0.0, < 2.12.0a0
+tensorflow-cpu >= 2.0.0, < 2.20.0a0
diff --git a/docs/source/conf.py b/docs/source/conf.py
index 5a8d6b3a..f0b8584b 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -135,3 +135,8 @@
 
 # If true, `todo` and `todoList` produce output, else they produce nothing.
 todo_include_todos = True
+
+# -- Options for copybutton extension ----------------------------------------
+
+# To make sphinx-copybutton skip all prompt characters generated by pygments
+copybutton_exclude = '.linenos, .gp'