This repository was archived by the owner on Jul 7, 2023. It is now read-only.
File tree Expand file tree Collapse file tree 3 files changed +13
-15
lines changed Expand file tree Collapse file tree 3 files changed +13
-15
lines changed Original file line number Diff line number Diff line change 7
7
services :
8
8
- docker
9
9
python :
10
- - " 2.7"
11
10
- " 3.6"
12
11
env :
13
12
global :
Original file line number Diff line number Diff line change @@ -20,7 +20,7 @@ t2t-trainer --problem=$T2T_PROBLEM --data_dir=$T2T_DATA_DIR --model=transformer
20
20
t2t-decoder --problem=$T2T_PROBLEM --data_dir=$T2T_DATA_DIR --model=transformer --hparams_set=transformer_tiny --output_dir=$T2T_TRAIN_DIR --decode_hparams=' num_samples=10'
21
21
22
22
# Test serving
23
- if [[ " $TRAVIS_PYTHON_VERSION " == " 2.7 " ]] && [[ " $ TF_VERSION" == " $TF_LATEST " ]]
23
+ if [[ " $TF_VERSION " == " $TF_LATEST " ]]
24
24
then
25
25
# Export for serving
26
26
pip install tensorflow_hub
Original file line number Diff line number Diff line change @@ -158,19 +158,7 @@ set_status
158
158
# set_status
159
159
# fi
160
160
161
- if [[ " $TRAVIS_PYTHON_VERSION " == " 2.7" ]] && [[ " $TF_VERSION " == " $TF_LATEST " ]]
162
- then
163
- # TODO(afrozm): Once we drop support for 1.10 we can get rid of this.
164
- pytest --disable-warnings \
165
- tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testTPUBeam
166
- set_status
167
- # TODO(afrozm): Enable other tests in the RL directory.
168
- # Can't add disable warning here since it parses flags.
169
- pytest tensor2tensor/rl/trainer_model_based_test.py
170
- set_status
171
- fi
172
-
173
- if [[ " $TRAVIS_PYTHON_VERSION " == " 3.6" ]] && [[ " $TF_VERSION " == " $TF_LATEST " ]]
161
+ if [[ " $TF_VERSION " == " $TF_LATEST " ]]
174
162
then
175
163
jupyter nbconvert --ExecutePreprocessor.kernel_name=python3 \
176
164
--ExecutePreprocessor.timeout=600 --to notebook --execute \
@@ -181,6 +169,17 @@ then
181
169
--ExecutePreprocessor.timeout=600 --to notebook --execute \
182
170
tensor2tensor/notebooks/t2t_problem.ipynb;
183
171
set_status
172
+
173
+ # TODO(afrozm): Once we drop support for 1.10 we can get rid of this.
174
+ pytest --disable-warnings \
175
+ tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testTPUBeam
176
+ set_status
177
+
178
+ # TODO(afrozm): Enable other tests in the RL directory.
179
+ # Can't add disable warning here since it parses flags.
180
+ pytest tensor2tensor/rl/trainer_model_based_test.py
181
+ set_status
182
+
184
183
fi
185
184
186
185
# Test --t2t_usr_dir
You can’t perform that action at this time.
0 commit comments