diff --git a/sdks/python/apache_beam/ml/inference/tensorflow_tests_requirements.txt b/sdks/python/apache_beam/ml/inference/tensorflow_tests_requirements.txt
index aebddf612263..e0a5c704de4f 100644
--- a/sdks/python/apache_beam/ml/inference/tensorflow_tests_requirements.txt
+++ b/sdks/python/apache_beam/ml/inference/tensorflow_tests_requirements.txt
@@ -19,4 +19,5 @@ keras<3.0.0
 tensorflow>=2.12.0
 tensorflow_hub>=0.10.0
 Pillow>=9.0.0
+typing-extensions>=4.8.0