Skip to content
This repository was archived by the owner on Jul 7, 2023. It is now read-only.

Commit 109987b

Browse files
afrozenatorCopybara-Service
authored andcommitted
Fast TPU decoding relies on an op added after TF 1.10, so disable that test for now.
PiperOrigin-RevId: 219231099
1 parent 1922524 commit 109987b

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

oss_scripts/oss_tests.sh

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,8 @@ pytest \
4747
--ignore=tensor2tensor/data_generators/allen_brain_test.py \
4848
--ignore=tensor2tensor/rl \
4949
--ignore=tensor2tensor/models/research \
50-
--deselect=tensor2tensor/layers/common_video_test.py::CommonVideoTest::testGifSummary
50+
--deselect=tensor2tensor/layers/common_video_test.py::CommonVideoTest::testGifSummary \
51+
--deselect=tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testTPUBeam
5152
set_status
5253

5354
pytest tensor2tensor/utils/registry_test.py
@@ -74,6 +75,9 @@ fi
7475

7576
if [[ "$TRAVIS_PYTHON_VERSION" == "2.7" ]] && [[ "$TF_VERSION" == "$TF_LATEST" ]]
7677
then
78+
# TODO(afrozm): Once we drop support for 1.10 we can get rid of this.
79+
pytest tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testTPUBeam
80+
set_status
7781
# TODO(afrozm): Enable other tests in the RL directory.
7882
pytest tensor2tensor/rl/trainer_model_based_test.py
7983
set_status

0 commit comments

Comments
 (0)