diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py
index b9016231b6a..599691bf100 100644
--- a/.circleci/create_circleci_config.py
+++ b/.circleci/create_circleci_config.py
@@ -199,7 +199,7 @@ pipelines_tf_job = CircleCIJob(
     "pipelines_tf",
     install_steps=[
         "pip install --upgrade pip",
-        "pip install .[sklearn,tf-cpu,testing,sentencepiece]",
+        "pip install .[sklearn,tf-cpu,testing,sentencepiece,vision]",
         "pip install tensorflow_probability",
     ],
     pytest_options={"rA": None},