- Disable the MuJoCo test from Travis since it needs installation, license and

setup.
 - Added a missing set_status from a previous change.
 - Disable overly verbose warnings summary from pytest.

PiperOrigin-RevId: 237319913

Author: afrozenator

oss_scripts/oss_tests.sh

@@ -43,13 +43,14 @@ set_status
 # trax tests need C++
 # others (see below) enable eager, so can't be tested along with the others in
 # pytest
-pytest \
+pytest --disable-warnings \
   --ignore=tensor2tensor/bin/t2t_trainer_test.py \
   --ignore=tensor2tensor/data_generators/algorithmic_math_test.py \
   --ignore=tensor2tensor/data_generators/allen_brain_test.py \
   --ignore=tensor2tensor/data_generators/ops/pack_sequences_ops_test.py \
   --ignore=tensor2tensor/data_generators/ops/subword_text_encoder_ops_test.py \
   --ignore=tensor2tensor/data_generators/problem_test.py \
+  --ignore=tensor2tensor/envs/mujoco_problems_test.py \
   --ignore=tensor2tensor/layers/bayes_test.py \
   --ignore=tensor2tensor/layers/common_attention_test.py \
   --ignore=tensor2tensor/layers/common_layers_test.py \
@@ -66,10 +67,10 @@ pytest \
   --ignore=tensor2tensor/models/video/nfg_uncond_test.py \
   --ignore=tensor2tensor/rl \
   --ignore=tensor2tensor/trax \
+  --ignore=tensor2tensor/utils/registry_test.py \
   --ignore=tensor2tensor/utils/t2t_model_test.py \
   --ignore=tensor2tensor/utils/test_utils.py \
   --ignore=tensor2tensor/utils/test_utils_test.py \
-  --ignore=tensor2tensor/utils/registry_test.py \
   --ignore=tensor2tensor/utils/trainer_lib_test.py \
   --ignore=tensor2tensor/visualization/visualization_test.py \
   --deselect=tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testTPUBeam
@@ -80,7 +81,7 @@ set_status
 # ImportError: /usr/lib/x86_64-linux-gnu/libstdc++.so.6: version `GLIBCXX_3.4.21' not found (required by /home/travis/virtualenv/python3.6.3/lib/python3.6/site-packages/jaxlib/_pywrap_xla.so)
 
 # These tests enable eager, so are tested separately.
-pytest \
+pytest --disable-warnings \
   tensor2tensor/data_generators/problem_test.py \
   tensor2tensor/layers/bayes_test.py \
   tensor2tensor/layers/common_attention_test.py \
@@ -94,17 +95,18 @@ pytest \
   tensor2tensor/utils/t2t_model_test.py \
   tensor2tensor/utils/test_utils_test.py \
   --deselect=tensor2tensor/layers/common_video_test.py::CommonVideoTest::testGifSummary
+set_status
 
-pytest tensor2tensor/utils/registry_test.py
+pytest --disable-warnings tensor2tensor/utils/registry_test.py
 set_status
 
-pytest tensor2tensor/utils/trainer_lib_test.py
+pytest --disable-warnings tensor2tensor/utils/trainer_lib_test.py
 set_status
 
-pytest tensor2tensor/visualization/visualization_test.py
+pytest --disable-warnings tensor2tensor/visualization/visualization_test.py
 set_status
 
-pytest tensor2tensor/data_generators/allen_brain_test.py
+pytest --disable-warnings tensor2tensor/data_generators/allen_brain_test.py
 set_status
 
 
@@ -113,17 +115,19 @@ if [[ "$TRAVIS_PYTHON_VERSION" == "2.7"  ]] && [[ "$TF_VERSION" == "tf-nightly"
 then
   # Ignores:
   # * Glow requires the CIFAR-10 dataset to be generated
-  pytest tensor2tensor/models/research \
+  pytest --disable-warnings tensor2tensor/models/research \
     --ignore=tensor2tensor/models/research/glow_test.py
   set_status
 fi
 
 if [[ "$TRAVIS_PYTHON_VERSION" == "2.7" ]] && [[ "$TF_VERSION" == "$TF_LATEST"  ]]
 then
     # TODO(afrozm): Once we drop support for 1.10 we can get rid of this.
-    pytest tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testTPUBeam
+    pytest --disable-warnings \
+      tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testTPUBeam
     set_status
     # TODO(afrozm): Enable other tests in the RL directory.
+    # Can't add disable warning here since it parses flags.
     pytest tensor2tensor/rl/trainer_model_based_test.py
     set_status
     jupyter nbconvert --ExecutePreprocessor.timeout=600 --to notebook --execute tensor2tensor/notebooks/hello_t2t.ipynb