Install vision for TF pipeline tests (#20771)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2022-12-15 11:16:37 +01:00 committed by GitHub
parent a9912d2fca
commit fe9152f67c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -199,7 +199,7 @@ pipelines_tf_job = CircleCIJob(
"pipelines_tf", "pipelines_tf",
install_steps=[ install_steps=[
"pip install --upgrade pip", "pip install --upgrade pip",
"pip install .[sklearn,tf-cpu,testing,sentencepiece]", "pip install .[sklearn,tf-cpu,testing,sentencepiece,vision]",
"pip install tensorflow_probability", "pip install tensorflow_probability",
], ],
pytest_options={"rA": None}, pytest_options={"rA": None},