travis_fold:start:worker_info
[0K[33;1mWorker information[0m
hostname: 1aa212aa-0013-4ec2-951f-dcb3dd56b699@1.worker-com-668464d7c8-q5cs4.gce-production-2
version: v6.2.17 https://github.com/travis-ci/worker/tree/bc420764f334fe0d096a417c80426f672f1b7d18
instance: travis-job-4dc0e17d-253a-4c6d-9850-3fd1ef4da5e3 travis-ci-sardonyx-xenial-1553530528-f909ac5 (via amqp)
startup: 6.280858032s
travis_fold:end:worker_info
[0Ktravis_time:start:2f915232
[0Ktravis_time:end:2f915232:start=1595218548858847774,finish=1595218548986060525,duration=127212751,event=no_world_writable_dirs
[0Ktravis_time:start:0ae5289c
[0Ktravis_time:end:0ae5289c:start=1595218548988989794,finish=1595218548995866071,duration=6876277,event=agent
[0Ktravis_time:start:02eebe4e
[0Ktravis_time:end:02eebe4e:start=1595218548998343551,finish=1595218549000124691,duration=1781140,event=check_unsupported
[0Ktravis_time:start:0dac8da4
[0Ktravis_fold:start:system_info
[0K[33;1mBuild system information[0m
Build language: python
Build dist: xenial
Build id: 176254524
Job id: 362633408
Runtime kernel version: 4.15.0-1028-gcp
travis-build version: 01e86bea
[34m[1mBuild image provisioning date and time[0m
Mon Mar 25 16:43:24 UTC 2019
[34m[1mOperating System Details[0m
Distributor ID:	Ubuntu
Description:	Ubuntu 16.04.6 LTS
Release:	16.04
Codename:	xenial
[34m[1mSystemd Version[0m
systemd 229
[34m[1mCookbooks Version[0m
42e42e4 https://github.com/travis-ci/travis-cookbooks/tree/42e42e4
[34m[1mgit version[0m
git version 2.21.0
[34m[1mbash version[0m
GNU bash, version 4.3.48(1)-release (x86_64-pc-linux-gnu)
[34m[1mgcc version[0m
gcc (Ubuntu 5.4.0-6ubuntu1~16.04.11) 5.4.0 20160609
[34m[1mdocker version[0m
Client:
 Version:           18.06.0-ce
 API version:       1.38
 Go version:        go1.10.3
 Git commit:        0ffa825
 Built:             Wed Jul 18 19:11:02 2018
 OS/Arch:           linux/amd64
 Experimental:      false

Server:
 Engine:
  Version:          18.06.0-ce
  API version:      1.38 (minimum version 1.12)
  Go version:       go1.10.3
  Git commit:       0ffa825
  Built:            Wed Jul 18 19:09:05 2018
  OS/Arch:          linux/amd64
  Experimental:     false
[34m[1mclang version[0m
clang version 7.0.0 (tags/RELEASE_700/final)
[34m[1mjq version[0m
jq-1.5
[34m[1mbats version[0m
Bats 0.4.0
[34m[1mshellcheck version[0m
0.6.0
[34m[1mshfmt version[0m
v2.6.3
[34m[1mccache version[0m
3.2.4
[34m[1mcmake version[0m
cmake version 3.12.4
[34m[1mheroku version[0m
heroku/7.22.7 linux-x64 node-v11.10.1
[34m[1mimagemagick version[0m
Version: ImageMagick 6.8.9-9 Q16 x86_64 2018-09-28 http://www.imagemagick.org
[34m[1mmd5deep version[0m
4.4
[34m[1mmercurial version[0m
version 4.8
[34m[1mmysql version[0m
mysql  Ver 14.14 Distrib 5.7.25, for Linux (x86_64) using  EditLine wrapper
[34m[1mopenssl version[0m
OpenSSL 1.0.2g  1 Mar 2016
[34m[1mpacker version[0m
1.3.3
[34m[1mpostgresql client version[0m
psql (PostgreSQL) 10.7 (Ubuntu 10.7-1.pgdg16.04+1)
[34m[1mragel version[0m
Ragel State Machine Compiler version 6.8 Feb 2013
[34m[1msudo version[0m
1.8.16
[34m[1mgzip version[0m
gzip 1.6
[34m[1mzip version[0m
Zip 3.0
[34m[1mvim version[0m
VIM - Vi IMproved 7.4 (2013 Aug 10, compiled Nov 24 2016 16:44:48)
[34m[1miptables version[0m
iptables v1.6.0
[34m[1mcurl version[0m
curl 7.47.0 (x86_64-pc-linux-gnu) libcurl/7.47.0 GnuTLS/3.4.10 zlib/1.2.8 libidn/1.32 librtmp/2.3
[34m[1mwget version[0m
GNU Wget 1.17.1 built on linux-gnu.
[34m[1mrsync version[0m
rsync  version 3.1.1  protocol version 31
[34m[1mgimme version[0m
v1.5.3
[34m[1mnvm version[0m
0.34.0
[34m[1mperlbrew version[0m
/home/travis/perl5/perlbrew/bin/perlbrew  - App::perlbrew/0.86
[34m[1mphpenv version[0m
rbenv 1.1.2
[34m[1mrvm version[0m
rvm 1.29.7 (latest) by Michal Papis, Piotr Kuczynski, Wayne E. Seguin [https://rvm.io]
[34m[1mdefault ruby version[0m
ruby 2.5.3p105 (2018-10-18 revision 65156) [x86_64-linux]
[34m[1mCouchDB version[0m
couchdb 1.6.1
[34m[1mElasticSearch version[0m
5.5.0
[34m[1mInstalled Firefox version[0m
firefox 63.0.1
[34m[1mMongoDB version[0m
MongoDB 4.0.7
[34m[1mPhantomJS version[0m
2.1.1
[34m[1mPre-installed PostgreSQL versions[0m
9.4.21
9.5.16
9.6.12
[34m[1mRedis version[0m
redis-server 5.0.4
[34m[1mPre-installed Go versions[0m
1.11.1
[34m[1mmvn version[0m
Apache Maven 3.6.0 (97c98ec64a1fdfee7767ce5ffb20918da4f719f3; 2018-10-24T18:41:47Z)
[34m[1mgradle version[0m
Gradle 4.10.2!
[34m[1mlein version[0m
Leiningen 2.9.1 on Java 11.0.2 OpenJDK 64-Bit Server VM
[34m[1mPre-installed Node.js versions[0m
v10.15.3
v11.0.0
v4.9.1
v6.17.0
v8.12.0
v8.15.1
v8.9
[34m[1mphpenv versions[0m
  system
  5.6
  5.6.40
  7.1
  7.1.27
  7.2
* 7.2.15 (set by /home/travis/.phpenv/version)
  hhvm
  hhvm-stable
[34m[1mcomposer --version[0m
Composer version 1.8.4 2019-02-11 10:52:10
[34m[1mPre-installed Ruby versions[0m
ruby-2.3.8
ruby-2.4.5
ruby-2.5.3
travis_fold:end:system_info
[0K
travis_time:end:0dac8da4:start=1595218549002576742,finish=1595218549008878487,duration=6301745,event=show_system_info
[0Ktravis_time:start:025df78a
[0Ktravis_time:end:025df78a:start=1595218549011527034,finish=1595218549022472923,duration=10945889,event=rm_riak_source
[0Ktravis_time:start:08576fe8
[0Ktravis_time:end:08576fe8:start=1595218549025211302,finish=1595218549030170686,duration=4959384,event=fix_rwky_redis
[0Ktravis_time:start:0d08f2bd
[0Ktravis_time:end:0d08f2bd:start=1595218549032905461,finish=1595218549650663872,duration=617758411,event=wait_for_network
[0Ktravis_time:start:00d70f46
[0Ktravis_time:end:00d70f46:start=1595218549653361641,finish=1595218549840716649,duration=187355008,event=update_apt_keys
[0Ktravis_time:start:0b4f64bc
[0Ktravis_time:end:0b4f64bc:start=1595218549843353914,finish=1595218549889734469,duration=46380555,event=fix_hhvm_source
[0Ktravis_time:start:034fdda2
[0Ktravis_time:end:034fdda2:start=1595218549892361979,finish=1595218549894547776,duration=2185797,event=update_mongo_arch
[0Ktravis_time:start:0fc1ac1e
[0Ktravis_time:end:0fc1ac1e:start=1595218549897019854,finish=1595218549933530661,duration=36510807,event=fix_sudo_enabled_trusty
[0Ktravis_time:start:053859c0
[0Ktravis_time:end:053859c0:start=1595218549936192239,finish=1595218549938031456,duration=1839217,event=update_glibc
[0Ktravis_time:start:062dc8b8
[0Ktravis_time:end:062dc8b8:start=1595218549940560080,finish=1595218549947033713,duration=6473633,event=clean_up_path
[0Ktravis_time:start:13e49af0
[0Ktravis_time:end:13e49af0:start=1595218549949498906,finish=1595218549956338237,duration=6839331,event=fix_resolv_conf
[0Ktravis_time:start:2f677f80
[0Ktravis_time:end:2f677f80:start=1595218549958843273,finish=1595218549966357251,duration=7513978,event=fix_etc_hosts
[0Ktravis_time:start:0c891b40
[0Ktravis_time:end:0c891b40:start=1595218549968801481,finish=1595218549977843567,duration=9042086,event=fix_mvn_settings_xml
[0Ktravis_time:start:0189af76
[0Ktravis_time:end:0189af76:start=1595218549980514327,finish=1595218549988211366,duration=7697039,event=no_ipv6_localhost
[0Ktravis_time:start:011b093e
[0Ktravis_time:end:011b093e:start=1595218549990665917,finish=1595218549992566032,duration=1900115,event=fix_etc_mavenrc
[0Ktravis_time:start:0b0643f0
[0Ktravis_time:end:0b0643f0:start=1595218549995152025,finish=1595218549997666408,duration=2514383,event=fix_wwdr_certificate
[0Ktravis_time:start:05e05354
[0Ktravis_time:end:05e05354:start=1595218550000173485,finish=1595218550019650378,duration=19476893,event=put_localhost_first
[0Ktravis_time:start:052aaa80
[0Ktravis_time:end:052aaa80:start=1595218550022198715,finish=1595218550024646294,duration=2447579,event=home_paths
[0Ktravis_time:start:00f1377c
[0Ktravis_time:end:00f1377c:start=1595218550027387535,finish=1595218550037488145,duration=10100610,event=disable_initramfs
[0Ktravis_time:start:0e991660
[0Ktravis_time:end:0e991660:start=1595218550040027656,finish=1595218550266900827,duration=226873171,event=disable_ssh_roaming
[0Ktravis_time:start:0e351cca
[0Ktravis_time:end:0e351cca:start=1595218550269699216,finish=1595218550271689073,duration=1989857,event=debug_tools
[0Ktravis_time:start:065fd4ad
[0Ktravis_time:end:065fd4ad:start=1595218550274328475,finish=1595218550276964220,duration=2635745,event=uninstall_oclint
[0Ktravis_time:start:10c97bbb
[0Ktravis_time:end:10c97bbb:start=1595218550279618244,finish=1595218550282173290,duration=2555046,event=rvm_use
[0Ktravis_time:start:02f7ea2a
[0Ktravis_time:end:02f7ea2a:start=1595218550284704788,finish=1595218550291231067,duration=6526279,event=rm_etc_boto_cfg
[0Ktravis_time:start:1a396a42
[0Ktravis_time:end:1a396a42:start=1595218550293680579,finish=1595218550296038712,duration=2358133,event=rm_oraclejdk8_symlink
[0Ktravis_time:start:32fe769e
[0Ktravis_time:end:32fe769e:start=1595218550298592480,finish=1595218550387564540,duration=88972060,event=enable_i386
[0Ktravis_time:start:18198e48
[0Ktravis_time:end:18198e48:start=1595218550390330346,finish=1595218550395670680,duration=5340334,event=update_rubygems
[0Ktravis_time:start:21a1396c
[0Ktravis_time:end:21a1396c:start=1595218550398221195,finish=1595218551126140066,duration=727918871,event=ensure_path_components
[0Ktravis_time:start:1bdaabfd
[0Ktravis_time:end:1bdaabfd:start=1595218551129249059,finish=1595218551131341983,duration=2092924,event=redefine_curl
[0Ktravis_time:start:004e7f7f
[0Ktravis_time:end:004e7f7f:start=1595218551134047273,finish=1595218551136079177,duration=2031904,event=nonblock_pipe
[0Ktravis_time:start:064716d2
[0Ktravis_time:end:064716d2:start=1595218551138720464,finish=1595218557169525414,duration=6030804950,event=apt_get_update
[0Ktravis_time:start:2506d2cc
[0Ktravis_time:end:2506d2cc:start=1595218557172786415,finish=1595218557174977246,duration=2190831,event=deprecate_xcode_64
[0Ktravis_time:start:06993850
[0Ktravis_time:end:06993850:start=1595218557177704843,finish=1595218559521274080,duration=2343569237,event=update_heroku
[0Ktravis_time:start:37133318
[0Ktravis_time:end:37133318:start=1595218559524213427,finish=1595218559526244054,duration=2030627,event=shell_session_update
[0Ktravis_time:start:02455372
[0Ktravis_fold:start:docker_mtu
[0Ktravis_fold:end:docker_mtu
[0Ktravis_time:end:02455372:start=1595218559528815603,finish=1595218561754432659,duration=2225617056,event=set_docker_mtu
[0Ktravis_time:start:10e8b8f9
[0Ktravis_fold:start:resolvconf
[0Ktravis_fold:end:resolvconf
[0Ktravis_time:end:10e8b8f9:start=1595218561757797588,finish=1595218561811261281,duration=53463693,event=resolvconf
[0Ktravis_time:start:058167b0
[0Ktravis_time:end:058167b0:start=1595218561815552453,finish=1595218561943235866,duration=127683413,event=maven_central_mirror
[0Ktravis_time:start:00a22b28
[0Ktravis_time:end:00a22b28:start=1595218561946227798,finish=1595218562030940710,duration=84712912,event=maven_https
[0Ktravis_time:start:04b05366
[0Ktravis_fold:start:services
[0Ktravis_time:start:03514d3c
[0K$ sudo systemctl start docker
travis_time:end:03514d3c:start=1595218562042854089,finish=1595218562052115801,duration=9261712,event=prepare
[0Ktravis_fold:end:services
[0Ktravis_time:end:03514d3c:start=1595218562042854089,finish=1595218565055901422,duration=3013047333,event=services
[0Ktravis_time:start:194db5a8
[0Ktravis_time:end:194db5a8:start=1595218565058961195,finish=1595218565060951756,duration=1990561,event=fix_ps4
[0Ktravis_time:start:0648b5f2
[0K
travis_fold:start:git.checkout
[0Ktravis_time:start:000a5c42
[0K$ git clone --depth=3 --branch=master --quiet https://github.com/aryaman4/trax.git aryaman4/trax
travis_time:end:000a5c42:start=1595218565066926506,finish=1595218566017029361,duration=950102855,event=checkout
[0K$ cd aryaman4/trax
$ git checkout -qf 0d99a51c0fbbd3faea1d344dbbce1912708e8096
travis_fold:end:git.checkout
[0K
travis_time:end:000a5c42:start=1595218565066926506,finish=1595218566044399333,duration=977472827,event=checkout
[0Ktravis_time:start:01ae8851
[0K
[33;1mSetting environment variables from .travis.yml[0m
$ export TF_LATEST="2.2.*"
$ export TF_VERSION="2.2.*"

travis_time:end:01ae8851:start=1595218566047246356,finish=1595218566052753641,duration=5507285,event=env
[0Ktravis_time:start:068ca27f
[0K$ source ~/virtualenv/python3.6/bin/activate
travis_time:end:068ca27f:start=1595218566055734401,finish=1595218566061379238,duration=5644837,event=
[0Ktravis_fold:start:cache.1
[0KSetting up build cache
$ export CASHER_DIR=${TRAVIS_HOME}/.casher
travis_time:start:13831526
[0K$ Installing caching utilities
travis_time:end:13831526:start=1595218566745967283,finish=1595218567015833192,duration=269865909,event=setup_casher
[0Ktravis_time:start:0f2ae05b
[0Ktravis_time:end:0f2ae05b:start=1595218567020162615,finish=1595218567022516532,duration=2353917,event=setup_casher
[0Ktravis_time:start:110d2774
[0Kattempting to download cache archive[0m
[32;1mfetching master/cache--linux-xenial-a6812ccfb4dc0afee8dfe70bac496b870a786fe4512527e931715ec7de567dfe--python-3.6.tgz[0m
[32;1mfetching master/cache-linux-xenial-a6812ccfb4dc0afee8dfe70bac496b870a786fe4512527e931715ec7de567dfe--python-3.6.tgz[0m
[32;1mfetching master/cache--python-3.6.tgz[0m
[31;1mcould not download cache[0m
travis_time:end:110d2774:start=1595218567025558518,finish=1595218568272653098,duration=1247094580,event=setup_casher
[0Ktravis_fold:end:cache.1
[0K
travis_fold:start:cache.pip
[0K
travis_time:start:025d2080
[0Ktravis_time:end:025d2080:start=1595218568276446890,finish=1595218568278875903,duration=2429013,event=setup_cache
[0Ktravis_time:start:0bbd5bbd
[0Kadding /home/travis/.cache/pip to cache[0m
travis_time:end:0bbd5bbd:start=1595218568281968219,finish=1595218569235238182,duration=953269963,event=setup_cache
[0Ktravis_fold:end:cache.pip
[0K$ python --version
Python 3.6.7
$ pip --version
pip 19.0.3 from /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages/pip (python 3.6)
travis_fold:start:install
[0Ktravis_time:start:01519dee
[0K$ ./oss_scripts/oss_pip_install.sh
set -e  # fail and exit on any command erroring

: "${TF_VERSION:?}"

# Make sure we have the latest pip and setuptools installed.
pip install -q -U pip
pip install -q -U setuptools

# Make sure we have the latest version of numpy - avoid problems we were
# seeing with Python 3
pip install -q -U numpy

# Install appropriate version to tensorflow.
if [[ "$TF_VERSION" == "tf-nightly"  ]]
then
  pip install tf-nightly;
else
  pip install -q "tensorflow==$TF_VERSION"
fi

# Just print the version again to make sure.
python -c 'import tensorflow as tf; print(tf.__version__)'
2.2.0

# First ensure that the base dependencies are sufficient for a full import
pip install -q -e .

# Then install the test dependencies
pip install -q -e .[tests]
# Make sure to install the atari extras for gym
pip install "gym[atari]"
Requirement already satisfied: gym[atari] in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (0.17.2)
Requirement already satisfied: numpy>=1.10.4 in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (from gym[atari]) (1.19.0)
Requirement already satisfied: cloudpickle<1.4.0,>=1.2.0 in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (from gym[atari]) (1.3.0)
Requirement already satisfied: pyglet<=1.5.0,>=1.4.0 in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (from gym[atari]) (1.5.0)
Requirement already satisfied: scipy in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (from gym[atari]) (1.4.1)
Requirement already satisfied: opencv-python; extra == "atari" in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (from gym[atari]) (4.3.0.36)
Requirement already satisfied: Pillow; extra == "atari" in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (from gym[atari]) (7.2.0)
Collecting atari-py~=0.2.0; extra == "atari"
  Downloading atari_py-0.2.6-cp36-cp36m-manylinux1_x86_64.whl (2.8 MB)
[?25l
[?25hRequirement already satisfied: future in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (from pyglet<=1.5.0,>=1.4.0->gym[atari]) (0.18.2)
Requirement already satisfied: six in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (from atari-py~=0.2.0; extra == "atari"->gym[atari]) (1.15.0)
Installing collected packages: atari-py
Successfully installed atari-py-0.2.6

# Coverage.
pip install coverage coveralls
Collecting coverage
  Downloading coverage-5.2-cp36-cp36m-manylinux1_x86_64.whl (229 kB)
[?25l
[?25hCollecting coveralls
  Downloading coveralls-2.1.1-py2.py3-none-any.whl (13 kB)
Requirement already satisfied: requests>=1.0.0 in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (from coveralls) (2.24.0)
Collecting docopt>=0.6.1
  Downloading docopt-0.6.2.tar.gz (25 kB)
Requirement already satisfied: certifi>=2017.4.17 in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (from requests>=1.0.0->coveralls) (2018.10.15)
Requirement already satisfied: idna<3,>=2.5 in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (from requests>=1.0.0->coveralls) (2.10)
Requirement already satisfied: chardet<4,>=3.0.2 in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (from requests>=1.0.0->coveralls) (3.0.4)
Requirement already satisfied: urllib3!=1.25.0,!=1.25.1,<1.26,>=1.21.1 in /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages (from requests>=1.0.0->coveralls) (1.25.9)
Building wheels for collected packages: docopt
  Building wheel for docopt (setup.py) ... [?25l- done
[?25h  Created wheel for docopt: filename=docopt-0.6.2-py2.py3-none-any.whl size=13703 sha256=07ecaddaec0df8d2adec6f5a3185cd11d1b0021a81ec9e47aa64f814a537a256
  Stored in directory: /home/travis/.cache/pip/wheels/3f/2a/fa/4d7a888e69774d5e6e855d190a8a51b357d77cc05eb1c097c9
Successfully built docopt
Installing collected packages: coverage, docopt, coveralls
Successfully installed coverage-5.2 coveralls-2.1.1 docopt-0.6.2
travis_time:end:01519dee:start=1595218569985718028,finish=1595218999015808709,duration=429030090681,event=install
[0Ktravis_fold:end:install
[0Ktravis_time:start:0b4e31ca
[0K$ ./oss_scripts/oss_tests.sh

# aliases aren't expanded in non-interactive shells by default.
shopt -s expand_aliases

# Instead of exiting on any failure with "set -e", we'll call set_status after
# each command and exit $STATUS at the end.
STATUS=0
function set_status() {
    local last_status=$?
    if [[ $last_status -ne 0 ]]
    then
      echo "<<<<<<FAILED>>>>>> Exit code: $last_status"
    fi
    STATUS=$(($last_status || $STATUS))
}

# Check env vars set
echo "${TF_VERSION:?}" && \
echo "${TF_LATEST:?}" && \
echo "${TRAVIS_PYTHON_VERSION:?}"
2.2.*
2.2.*
3.6
set_status
if [[ $STATUS -ne 0 ]]
then
  exit $STATUS
fi

# Check import.
python -c "import trax"
set_status

# Run pytest with coverage.
alias pytest='coverage run -m pytest'

# Check tests, separate out directories for easy triage.

## Core Trax and Supervised Learning

# Disabled test fails with "disable-warnings", tested separately. These tests
# parse flags, so it is important that no other flags be given to pytest for
# these.

pytest --disable-warnings \
  --ignore=trax/supervised/training_test.py \
  trax/supervised
[1m============================= test session starts ==============================[0m
platform linux -- Python 3.6.7, pytest-4.3.1, py-1.7.0, pluggy-0.8.0
rootdir: /home/travis/build/aryaman4/trax, inifile: pytest.ini
plugins: typeguard-2.9.1
[1mcollecting ... [0m[1m
collecting 5 items                                                             [0m[1m
collected 55 items                                                             [0m

trax/supervised/decoding_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[33ms[0m[36m                                   [  9%][0m
trax/supervised/inputs_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                           [ 36%][0m
trax/supervised/lr_schedules_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                [ 43%][0m
trax/supervised/mnist_test.py [32m.[0m[36m                                          [ 45%][0m
trax/supervised/tf_inputs_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[33ms[0m[32m.[0m[36m                          [ 69%][0m
trax/supervised/trainer_lib_test.py [32m.[0m[32m.[0m[32m.[0m[33ms[0m[33ms[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[33ms[0m[32m.[0m[32m.[0m[36m                    [100%][0m

============================ slowest test durations ============================
190.16s call     trax/supervised/mnist_test.py::MnistTest::test_train_mnist
98.58s call     trax/supervised/trainer_lib_test.py::TraxTest::test_no_int32_or_uint32_returned
32.81s call     trax/supervised/tf_inputs_test.py::TFInputsTest::test_inputs_using_generic_text_dataset_preprocess_fn
11.26s call     trax/supervised/decoding_test.py::DecodingTest::test_autoregressive_sample_transformer_quality
8.20s call     trax/supervised/trainer_lib_test.py::TraxTest::test_train_eval_predict0
7.63s call     trax/supervised/decoding_test.py::DecodingTest::test_autoregressive_sample_reformerlm
6.81s call     trax/supervised/trainer_lib_test.py::TraxTest::test_train_restart1
6.65s call     trax/supervised/trainer_lib_test.py::TraxTest::test_train_restart0
5.48s call     trax/supervised/decoding_test.py::DecodingTest::test_autoregressive_sample_transformer
4.94s call     trax/supervised/trainer_lib_test.py::TraxTest::test_reset_twice0
4.77s call     trax/supervised/tf_inputs_test.py::TFInputsTest::test_c4_bare_preprocess_fn_denoising_objective
4.73s call     trax/supervised/trainer_lib_test.py::TraxTest::test_train_eval_predict_sm31
4.51s call     trax/supervised/trainer_lib_test.py::TraxTest::test_train_eval_predict1
4.27s call     trax/supervised/trainer_lib_test.py::TraxTest::test_train_with_weights0
4.24s call     trax/supervised/trainer_lib_test.py::TraxTest::test_train_eval_predict_sm30
3.78s call     trax/supervised/trainer_lib_test.py::TraxTest::test_train_with_weights1
3.52s call     trax/supervised/decoding_test.py::DecodingTest::test_autoregressive_sample_transformerlm
2.77s call     trax/supervised/trainer_lib_test.py::TraxTest::test_reset_twice1
1.46s call     trax/supervised/tf_inputs_test.py::TFInputsTest::test_c4_bare_preprocess_fn
0.97s call     trax/supervised/tf_inputs_test.py::TFInputsTest::test_generic_text_dataset_preprocess_fn
0.69s call     trax/supervised/tf_inputs_test.py::TFInputsTest::test_c4_preprocess
0.36s call     trax/supervised/tf_inputs_test.py::TFInputsTest::test_truncate_dataset_on_len
0.31s call     trax/supervised/tf_inputs_test.py::TFInputsTest::test_pad_dataset_to_length
0.30s call     trax/supervised/tf_inputs_test.py::TFInputsTest::test_filter_dataset_on_len
0.23s call     trax/supervised/tf_inputs_test.py::TFInputsTest::test_lm_token_preprocessing
0.01s call     trax/supervised/tf_inputs_test.py::TFInputsTest::test_get_t5_preprocessor_by_name
0.01s call     trax/supervised/trainer_lib_test.py::EpochsTest::test_cuts_epoch_when_total_steps_reached
0.01s call     trax/supervised/tf_inputs_test.py::TFInputsTest::test_c4_pretrain

(0.00 durations hidden.  Use -vv to show these durations.)
[33m[1m============ 50 passed, 5 skipped, 10537 warnings in 414.70 seconds ============[0m
set_status

pytest trax/supervised/training_test.py
[1m============================= test session starts ==============================[0m
platform linux -- Python 3.6.7, pytest-4.3.1, py-1.7.0, pluggy-0.8.0
rootdir: /home/travis/build/aryaman4/trax, inifile: pytest.ini
plugins: typeguard-2.9.1
[1mcollecting ... [0m[1m
collecting 6 items                                                             [0m[1m
collected 6 items                                                              [0m

trax/supervised/training_test.py [32m.[0m[33ms[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                  [100%][0m

[33m=============================== warnings summary ===============================[0m
trax/supervised/training_test.py::TrainingTest::test_loop_no_eval_task
  /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages/jax/lib/xla_bridge.py:125: UserWarning: No GPU/TPU found, falling back to CPU.
    warnings.warn('No GPU/TPU found, falling back to CPU.')

-- Docs: https://docs.pytest.org/en/latest/warnings.html
============================ slowest test durations ============================
2.27s call     trax/supervised/training_test.py::TrainingTest::test_summaries_are_written
0.61s call     trax/supervised/training_test.py::TrainingTest::test_loop_no_eval_task
0.30s call     trax/supervised/training_test.py::TrainingTest::test_train_dense_layer_with_momentum
0.22s call     trax/supervised/training_test.py::TrainingTest::test_train_dense_layer
0.13s call     trax/supervised/training_test.py::TrainingTest::test_train_dense_layer_evals

(0.00 durations hidden.  Use -vv to show these durations.)
[33m[1m=============== 5 passed, 1 skipped, 1 warnings in 8.29 seconds ================[0m
set_status

pytest --disable-warnings \
  --ignore=trax/layers/initializers_test.py \
  trax/layers
[1m============================= test session starts ==============================[0m
platform linux -- Python 3.6.7, pytest-4.3.1, py-1.7.0, pluggy-0.8.0
rootdir: /home/travis/build/aryaman4/trax, inifile: pytest.ini
plugins: typeguard-2.9.1
[1mcollecting ... [0m[1m
collecting 2 items                                                             [0m[1m
collecting 156 items                                                           [0m[1m
collected 170 items                                                            [0m

trax/layers/acceleration_test.py [32m.[0m[32m.[0m[36m                                      [  1%][0m
trax/layers/activation_fns_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                 [  4%][0m
trax/layers/attention_test.py [32m.[0m[32m.[0m[32m.[0m[36m                                        [  5%][0m
trax/layers/base_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                  [ 14%][0m
trax/layers/combinators_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m [ 37%]
[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                                                [ 42%][0m
trax/layers/convolution_test.py [32m.[0m[32m.[0m[32m.[0m[36m                                      [ 44%][0m
trax/layers/core_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                          [ 57%][0m
trax/layers/metrics_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                    [ 62%][0m
trax/layers/normalization_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                         [ 71%][0m
trax/layers/pooling_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                   [ 77%][0m
trax/layers/reversible_test.py [32m.[0m[32m.[0m[36m                                        [ 78%][0m
trax/layers/rnn_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                            [ 81%][0m
trax/layers/research/efficient_attention_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[33ms[0m[36m             [ 87%][0m
trax/layers/research/position_encodings_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m    [100%][0m

============================ slowest test durations ============================
73.07s call     trax/layers/research/efficient_attention_test.py::EfficientAttentionTest::test_batching_lsh_self_attention
40.82s call     trax/layers/research/efficient_attention_test.py::EfficientAttentionTest::test_fast_inference_self_attention
25.48s call     trax/layers/research/efficient_attention_test.py::EfficientAttentionTest::test_batching_self_attention
22.59s call     trax/layers/research/efficient_attention_test.py::EfficientAttentionTest::test_lsh_determinism_causal
19.43s call     trax/layers/research/efficient_attention_test.py::EfficientAttentionTest::test_lsh_determinism_non_causal
7.80s call     trax/layers/research/efficient_attention_test.py::EfficientAttentionTest::test_lsh_self_attention_masked_non_causal
3.29s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_0::test_inference_0
1.93s call     trax/layers/research/efficient_attention_test.py::EfficientAttentionTest::test_lsh_self_attention
1.88s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_0::test_inference_1
1.60s call     trax/layers/rnn_test.py::RnnTest::test_conv_gru_cell
1.40s call     trax/layers/rnn_test.py::RnnTest::test_sru
1.28s call     trax/layers/rnn_test.py::RnnTest::test_gru_cell
1.26s call     trax/layers/research/efficient_attention_test.py::EfficientAttentionTest::test_self_attention
0.81s call     trax/layers/combinators_test.py::SerialTest::test_flat_weights_and_state
0.80s call     trax/layers/rnn_test.py::RnnTest::test_lstm_cell
0.70s call     trax/layers/convolution_test.py::ConvolutionTest::test_call
0.60s call     trax/layers/acceleration_test.py::AccelerationTest::test_accelerated_pad
0.59s call     trax/layers/core_test.py::LogGaussianTest::test_log_gaussian_pdf
0.55s call     trax/layers/normalization_test.py::BatchNormTest::test_forward_shape
0.54s call     trax/layers/convolution_test.py::CausalConvolutionTest::test_causal_conv
0.52s call     trax/layers/research/efficient_attention_test.py::EfficientAttentionTest::test_lsh_self_attention_tf
0.52s call     trax/layers/base_test.py::BaseLayerTest::test_custom_id_grad_jax
0.42s call     trax/layers/combinators_test.py::ParallelTest::test_shared_weights_for_shared_parallel
0.39s call     trax/layers/core_test.py::EmbeddingTest::test_new_weights
0.34s call     trax/layers/core_test.py::DropoutTest::test_call_in_train_mode
0.33s call     trax/layers/core_test.py::DenseTest::test_call_uses_and_caches_supplied_weights
0.32s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_2::test_inference_0
0.30s call     trax/layers/core_test.py::EmbeddingTest::test_forward
0.26s call     trax/layers/combinators_test.py::SerialTest::test_flat_weights_and_state_shared
0.26s call     trax/layers/metrics_test.py::MetricsTest::test_cross_entropy
0.23s call     trax/layers/metrics_test.py::MetricsTest::test_accuracy
0.23s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_0::test_inference_2
0.21s call     trax/layers/research/efficient_attention_test.py::EfficientAttentionTest::test_self_attention_tf
0.20s call     trax/layers/normalization_test.py::FilterResponseNormTest::test_forward_shape_learn_epsilon_false
0.20s call     trax/layers/core_test.py::DenseTest::test_call_no_bias
0.17s call     trax/layers/combinators_test.py::ParallelTest::test_shared_weights_nested
0.17s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_2::test_inference_2
0.15s call     trax/layers/normalization_test.py::LayerNormTest::test_forward_dtype_jax32
0.15s call     trax/layers/combinators_test.py::SerialTest::test_set_rng_recurse_two_levels
0.14s call     trax/layers/normalization_test.py::BatchNormTest::test_forward_dtype_jax32
0.14s call     trax/layers/combinators_test.py::SerialWithSideOutputsTest::test_serial_with_side_outputs_div_div
0.12s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_1::test_inference_2
0.11s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_1::test_inference_0
0.11s call     trax/layers/acceleration_test.py::AccelerationTest::test_accelerated_same_result
0.11s call     trax/layers/combinators_test.py::SerialTest::test_one_in_one_out
0.11s call     trax/layers/core_test.py::LogGaussianTest::test_log_gaussian_diag_pdf
0.10s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_1::test_inference_1
0.10s call     trax/layers/convolution_test.py::ConvolutionTest::test_call_rebatch
0.07s call     trax/layers/normalization_test.py::BatchNormTest::test_forward_momentum_800
0.07s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_2::test_inference_1
0.07s call     trax/layers/combinators_test.py::SerialTest::test_empty_list_is_no_op
0.07s call     trax/layers/pooling_test.py::AvgPoolTest::test_padding_same
0.07s call     trax/layers/combinators_test.py::ScanTest::test_multi_input
0.07s call     trax/layers/metrics_test.py::MetricsTest::test_weighted_mean_shape
0.06s call     trax/layers/attention_test.py::AttentionTest::test_shift_right
0.06s call     trax/layers/pooling_test.py::AvgPoolTest::test_forward_shape
0.06s call     trax/layers/combinators_test.py::ScanTest::test_axis_1
0.06s call     trax/layers/attention_test.py::AttentionTest::test_shift_right_float
0.05s call     trax/layers/pooling_test.py::AvgPoolTest::test_forward
0.05s call     trax/layers/base_test.py::BaseLayerTest::test_custom_zero_grad_jax
0.05s call     trax/layers/pooling_test.py::SumPoolTest::test_forward_shape
0.05s call     trax/layers/combinators_test.py::SerialTest::test_shared_weights_for_shared_serial
0.04s call     trax/layers/metrics_test.py::MetricsTest::test_accuracy_scalar
0.04s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_0::test_training_1
0.04s call     trax/layers/combinators_test.py::ScanTest::test_default_axis
0.04s call     trax/layers/metrics_test.py::MetricsTest::test_weighted_sequence_mean_semantics
0.03s call     trax/layers/combinators_test.py::ScanTest::test_no_carry
0.03s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_0::test_training_0
0.03s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_0::test_training_3
0.03s call     trax/layers/pooling_test.py::MaxPoolTest::test_forward_shape
0.03s call     trax/layers/normalization_test.py::BatchNormTest::test_forward_dtype_tf32
0.03s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_1::test_training_3
0.03s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_1::test_training_0
0.03s call     trax/layers/pooling_test.py::MaxPoolTest::test_padding_same
0.02s call     trax/layers/pooling_test.py::SumPoolTest::test_padding_same
0.02s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_1::test_training_1
0.02s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_0::test_training_2
0.02s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_1::test_training_2
0.02s call     trax/layers/activation_fns_test.py::ActivationFnsTest::test_hard_sigmoid
0.02s call     trax/layers/core_test.py::EmbeddingTest::test_explicit_kernel_initializer
0.02s call     trax/layers/normalization_test.py::FilterResponseNormTest::test_forward_shape_learn_epsilon_true
0.02s call     trax/layers/activation_fns_test.py::ActivationFnsTest::test_parametric_relu
0.01s call     trax/layers/pooling_test.py::MaxPoolTest::test_forward
0.01s call     trax/layers/pooling_test.py::SumPoolTest::test_forward
0.01s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_2::test_training_1
0.01s call     trax/layers/core_test.py::EmbeddingTest::test_large_inputs_clip_to_upper_bound
0.01s call     trax/layers/core_test.py::EmbeddingTest::test_negative_inputs_clip_to_zero
0.01s call     trax/layers/combinators_test.py::SerialTest::test_shared_weights_double_nested
0.01s call     trax/layers/metrics_test.py::MetricsTest::test_l2_loss
0.01s call     trax/layers/metrics_test.py::MetricsTest::test_weighted_mean_semantics
0.01s call     trax/layers/activation_fns_test.py::ActivationFnsTest::test_leaky_relu
0.01s call     trax/layers/pooling_test.py::MaxPoolTest::test_padding_default
0.01s call     trax/layers/base_test.py::FnTest::test_forward
0.01s call     trax/layers/combinators_test.py::ConcatenateTest::test_n_items_is_not_default
0.01s call     trax/layers/combinators_test.py::ConcatenateTest::test_axis_1
0.01s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_2::test_training_2
0.01s call     trax/layers/combinators_test.py::ConcatenateTest::test_axis_0
0.01s call     trax/layers/normalization_test.py::BatchNormTest::test_forward_momentum_900
0.01s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_2::test_training_3
0.01s call     trax/layers/combinators_test.py::SerialTest::test_shared_weights_nested
0.01s call     trax/layers/research/position_encodings_test.py::PositionEncodingsTest_2::test_training_0
0.01s call     trax/layers/normalization_test.py::BatchNormTest::test_forward_momentum_999
0.01s call     trax/layers/core_test.py::DenseTest::test_separate_instances_have_separate_weights
0.01s call     trax/layers/normalization_test.py::BatchNormTest::test_forward_dtype_tf64
0.01s call     trax/layers/core_test.py::DenseTest::test_shared_instance_means_shared_weights
0.01s call     trax/layers/metrics_test.py::MetricsTest::test_cross_entropy_loss
0.01s call     trax/layers/combinators_test.py::SerialTest::test_shared_weights
0.01s call     trax/layers/reversible_test.py::ReversibleLayerTest::test_reversible_swap_tf

(0.00 durations hidden.  Use -vv to show these durations.)
[33m[1m============= 169 passed, 1 skipped, 39 warnings in 220.56 seconds =============[0m
set_status

pytest trax/layers/initializers_test.py
[1m============================= test session starts ==============================[0m
platform linux -- Python 3.6.7, pytest-4.3.1, py-1.7.0, pluggy-0.8.0
rootdir: /home/travis/build/aryaman4/trax, inifile: pytest.ini
plugins: typeguard-2.9.1
[1mcollecting ... [0m[1m
collecting 10 items                                                            [0m[1m
collected 10 items                                                             [0m

trax/layers/initializers_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                              [100%][0m

[33m=============================== warnings summary ===============================[0m
trax/layers/initializers_test.py::InitializersTest::test_from_file
  /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages/jax/lib/xla_bridge.py:125: UserWarning: No GPU/TPU found, falling back to CPU.
    warnings.warn('No GPU/TPU found, falling back to CPU.')

-- Docs: https://docs.pytest.org/en/latest/warnings.html
============================ slowest test durations ============================
0.64s call     trax/layers/initializers_test.py::InitializersTest::test_orthogonal
0.41s call     trax/layers/initializers_test.py::InitializersTest::test_glorot_normal
0.38s call     trax/layers/initializers_test.py::InitializersTest::test_glorot_uniform
0.37s call     trax/layers/initializers_test.py::InitializersTest::test_random_uniform
0.02s call     trax/layers/initializers_test.py::InitializersTest::test_from_file
0.02s call     trax/layers/initializers_test.py::InitializersTest::test_kaiming_normal
0.01s call     trax/layers/initializers_test.py::InitializersTest::test_kaiming_uniform

(0.00 durations hidden.  Use -vv to show these durations.)
[33m[1m==================== 10 passed, 1 warnings in 6.65 seconds =====================[0m
set_status

pytest --disable-warnings trax/fastmath
[1m============================= test session starts ==============================[0m
platform linux -- Python 3.6.7, pytest-4.3.1, py-1.7.0, pluggy-0.8.0
rootdir: /home/travis/build/aryaman4/trax, inifile: pytest.ini
plugins: typeguard-2.9.1
[1mcollecting ... [0m[1m
collecting 5 items                                                             [0m[1m
collected 5 items                                                              [0m

trax/fastmath/ops_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[33ms[0m[36m                                          [100%][0m

============================ slowest test durations ============================
0.02s call     trax/fastmath/ops_test.py::BackendTest::test_numpy_backend_delegation

(0.00 durations hidden.  Use -vv to show these durations.)
[33m[1m=============== 4 passed, 1 skipped, 1 warnings in 4.78 seconds ================[0m
set_status

# Disabled tests are quasi integration tests.
pytest --disable-warnings \
  --ignore=trax/models/reformer/reformer_e2e_test.py \
  --ignore=trax/models/reformer/reformer_memory_test.py \
  --ignore=trax/models/reformer/reformer_oom_test.py \
  trax/models
[1m============================= test session starts ==============================[0m
platform linux -- Python 3.6.7, pytest-4.3.1, py-1.7.0, pluggy-0.8.0
rootdir: /home/travis/build/aryaman4/trax, inifile: pytest.ini
plugins: typeguard-2.9.1
[1mcollecting ... [0m[1m
collecting 2 items                                                             [0m[1m
collected 29 items                                                             [0m

trax/models/atari_cnn_test.py [32m.[0m[33ms[0m[32m.[0m[33ms[0m[36m                                       [ 13%][0m
trax/models/mlp_test.py [32m.[0m[32m.[0m[36m                                               [ 20%][0m
trax/models/neural_gpu_test.py [32m.[0m[36m                                         [ 24%][0m
trax/models/resnet_test.py [32m.[0m[32m.[0m[36m                                            [ 31%][0m
trax/models/rl_test.py [32m.[0m[32m.[0m[32m.[0m[36m                                               [ 41%][0m
trax/models/rnn_test.py [32m.[0m[32m.[0m[36m                                               [ 48%][0m
trax/models/transformer_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                              [ 86%][0m
trax/models/reformer/reformer_test.py [32m.[0m[32m.[0m[32m.[0m[36m                                [ 96%][0m
trax/models/research/skipping_transformer_test.py [32m.[0m[36m                      [100%][0m

============================ slowest test durations ============================
77.19s call     trax/models/reformer/reformer_test.py::ReformerTest::test_reformer_no_enc_dec_attention_one_step
44.67s call     trax/models/research/skipping_transformer_test.py::SkippingTransformerTest::test_skipping_transformer_forward_shape
32.85s call     trax/models/reformer/reformer_test.py::ReformerTest::test_reformer_lm_lsh
17.13s call     trax/models/resnet_test.py::ResnetTest::test_resnet
7.59s call     trax/models/transformer_test.py::TransformerTest::test_transformer_noencdec_forward_shape
6.74s call     trax/models/transformer_test.py::TransformerTest::test_transformer_forward_shape_different_size
5.32s call     trax/models/transformer_test.py::TransformerTest::test_dot_product_causal_attention_fast_inference
4.54s call     trax/models/neural_gpu_test.py::NeuralGPUTest::test_ngpu
3.16s call     trax/models/reformer/reformer_test.py::ReformerTest::test_reformer_lm_forward_shape
3.14s call     trax/models/atari_cnn_test.py::AtariCnnTest::test_computes
2.57s call     trax/models/rnn_test.py::RNNTest::test_grulm_forward_shape
2.22s call     trax/models/resnet_test.py::ResnetTest::test_wide_resnet
1.90s call     trax/models/transformer_test.py::TransformerTest::test_transformer_forward_shape_same_size
1.67s call     trax/models/mlp_test.py::MLPTest::test_mlp_forward_shape
1.59s call     trax/models/rnn_test.py::RNNTest::test_rnnlm_forward_shape
1.51s call     trax/models/transformer_test.py::TransformerTest::test_transformer_forward_shape_same_vocab
1.47s call     trax/models/mlp_test.py::MLPTest::test_pure_mlp_forward_shape
1.10s call     trax/models/atari_cnn_test.py::FrameStackMLPTest::test_computes
1.09s call     trax/models/transformer_test.py::TransformerTest::test_transformer_lm_forward_shape
0.49s call     trax/models/rl_test.py::RLTest::test_policy_and_value_forward_shape
0.09s call     trax/models/transformer_test.py::TransformerTest::test_concat_with_padding
0.09s call     trax/models/transformer_test.py::TransformerTest::test_strip_from_concatenate_with_padding
0.02s call     trax/models/transformer_test.py::TransformerTest::test_mask_of_right_shift
0.02s call     trax/models/transformer_test.py::TransformerTest::test_mask_of_right_shift_3dims
0.01s call     trax/models/rl_test.py::RLTest::test_policy_forward_shape
0.01s call     trax/models/rl_test.py::RLTest::test_value_forward_shape

(0.00 durations hidden.  Use -vv to show these durations.)
[33m[1m============== 27 passed, 2 skipped, 3 warnings in 223.23 seconds ==============[0m
set_status

pytest --disable-warnings trax/optimizers
[1m============================= test session starts ==============================[0m
platform linux -- Python 3.6.7, pytest-4.3.1, py-1.7.0, pluggy-0.8.0
rootdir: /home/travis/build/aryaman4/trax, inifile: pytest.ini
plugins: typeguard-2.9.1
[1mcollecting ... [0m[1m
collecting 1 item                                                              [0m[1m
collected 1 item                                                               [0m

trax/optimizers/optimizers_test.py [32m.[0m[36m                                     [100%][0m

============================ slowest test durations ============================

(0.00 durations hidden.  Use -vv to show these durations.)
[33m[1m===================== 1 passed, 1 warnings in 4.88 seconds =====================[0m
set_status

## RL Trax

# Disabled tests fail with "disable-warnings", tested separately.
pytest --disable-warnings \
  --ignore=trax/rl/actor_critic_joint_test.py \
  --ignore=trax/rl/actor_critic_test.py \
  --ignore=trax/rl/task_test.py \
  --ignore=trax/rl/training_test.py \
  trax/rl
[1m============================= test session starts ==============================[0m
platform linux -- Python 3.6.7, pytest-4.3.1, py-1.7.0, pluggy-0.8.0
rootdir: /home/travis/build/aryaman4/trax, inifile: pytest.ini
plugins: typeguard-2.9.1
[1mcollecting ... [0m[1m
collecting 22 items                                                            [0m[1m
collected 57 items                                                             [0m

trax/rl/advantages_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                        [ 38%][0m
trax/rl/distributions_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                      [ 47%][0m
trax/rl/normalization_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                       [ 54%][0m
trax/rl/serialization_utils_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                            [ 70%][0m
trax/rl/space_serializer_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[33ms[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[33ms[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[33ms[0m[32m.[0m[36m                       [100%][0m

============================ slowest test durations ============================
1.39s call     trax/rl/serialization_utils_test.py::SerializationTest::test_wrapped_policy_continuous_serialized
1.35s call     trax/rl/serialization_utils_test.py::SerializationTest::test_extract_inner_model
1.21s call     trax/rl/serialization_utils_test.py::SerializationTest::test_serialized_model_continuous
0.90s call     trax/rl/serialization_utils_test.py::SerializationTest::test_wrapped_policy_continuous_raw
0.33s call     trax/rl/distributions_test.py::DistributionsTest::test_shapes_multi_discrete
0.27s call     trax/rl/serialization_utils_test.py::SerializationTest::test_serialized_model_discrete
0.25s call     trax/rl/normalization_test.py::NormalizationTest::test_normalize_collect
0.24s call     trax/rl/space_serializer_test.py::BoxSpaceSerializerTest::test_bounds_space
0.21s call     trax/rl/distributions_test.py::DistributionsTest::test_shapes_discrete
0.17s call     trax/rl/space_serializer_test.py::BoxSpaceSerializerTest::test_commutes
0.08s call     trax/rl/distributions_test.py::DistributionsTest::test_gaussian_probability_sums_to_one_1d
0.05s call     trax/rl/distributions_test.py::DistributionsTest::test_shapes_gaussian
0.04s call     trax/rl/space_serializer_test.py::BoxSpaceSerializerTest::test_serializes_boundaries
0.03s call     trax/rl/space_serializer_test.py::BoxSpaceSerializerTest::test_representation_changes
0.03s call     trax/rl/distributions_test.py::DistributionsTest::test_gaussian_probability_sums_to_one_2d
0.02s call     trax/rl/space_serializer_test.py::BoxSpaceSerializerTest::test_significance_map
0.02s call     trax/rl/space_serializer_test.py::BoxSpaceSerializerTest::test_representation_length
0.01s call     trax/rl/space_serializer_test.py::MultiDiscreteSpaceSerializerTest::test_significance_map
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_gae_bias_decreases_with_gae_lambda_0.5_0.7
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_td_k_bias_decreases_with_k_2_3
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_td_k_variance_increases_with_k_2_3
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_td_k_bias_decreases_with_k_1_2
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_td_lambda_bias_decreases_with_lambda_0.5_0.7
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_td_k_variance_increases_with_k_1_2
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_td_lambda_bias_decreases_with_lambda_0.7_0.9
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_td_lambda_variance_increases_with_lambda_0.5_0.7
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_td_lambda_variance_increases_with_lambda_0.7_0.9
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_gae_variance_increases_with_gae_lambda_0.5_0.7
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_td_lambda_variance_lower_than_monte_carlo
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_gae_bias_decreases_with_gae_lambda_0.7_0.9
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_td_k_variance_lower_than_monte_carlo
0.01s call     trax/rl/advantages_test.py::AdvantagesTest::test_gae_variance_increases_with_gae_lambda_0.7_0.9

(0.00 durations hidden.  Use -vv to show these durations.)
[33m[1m============== 54 passed, 3 skipped, 1 warnings in 11.70 seconds ===============[0m
set_status

pytest \
  trax/rl/actor_critic_joint_test.py \
  trax/rl/actor_critic_test.py \
  trax/rl/task_test.py \
  trax/rl/training_test.py
[1m============================= test session starts ==============================[0m
platform linux -- Python 3.6.7, pytest-4.3.1, py-1.7.0, pluggy-0.8.0
rootdir: /home/travis/build/aryaman4/trax, inifile: pytest.ini
plugins: typeguard-2.9.1
[1mcollecting ... [0m[1m
collecting 6 items                                                             [0m[1m
collected 26 items                                                             [0m

trax/rl/actor_critic_joint_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                [ 23%][0m
trax/rl/actor_critic_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                   [ 57%][0m
trax/rl/task_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                           [ 92%][0m
trax/rl/training_test.py [32m.[0m[32m.[0m[36m                                              [100%][0m

[33m=============================== warnings summary ===============================[0m
trax/rl/actor_critic_joint_test.py::ActorCriticJointTest::test_awrjoint_save_restore
  /home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages/jax/lib/xla_bridge.py:125: UserWarning: No GPU/TPU found, falling back to CPU.
    warnings.warn('No GPU/TPU found, falling back to CPU.')

-- Docs: https://docs.pytest.org/en/latest/warnings.html
============================ slowest test durations ============================
45.74s call     trax/rl/training_test.py::TrainingTest::test_policytrainer_cartpole
40.27s call     trax/rl/actor_critic_joint_test.py::ActorCriticJointTest::test_jointa2ctrainer_cartpole_transformer
37.27s call     trax/rl/actor_critic_test.py::ActorCriticTest::test_sanity_awrtrainer_transformer_cartpole
33.23s call     trax/rl/actor_critic_test.py::ActorCriticTest::test_awrtrainer_cartpole_shared
31.80s call     trax/rl/actor_critic_test.py::ActorCriticTest::test_awrtrainer_cartpole
30.68s call     trax/rl/actor_critic_joint_test.py::ActorCriticJointTest::test_jointawrtrainer_cartpole_transformer
10.86s call     trax/rl/actor_critic_test.py::ActorCriticTest::test_sanity_ppo_cartpole
10.12s call     trax/rl/actor_critic_test.py::ActorCriticTest::test_sampling_awrtrainer_cartpole
8.87s call     trax/rl/actor_critic_test.py::ActorCriticTest::test_sampling_awrtrainer_mountain_acr
8.76s call     trax/rl/actor_critic_test.py::ActorCriticTest::test_sampling_awrtrainer_cartpole_sample_all_discrete
8.29s call     trax/rl/training_test.py::TrainingTest::test_policytrainer_save_restore
7.99s call     trax/rl/actor_critic_test.py::ActorCriticTest::test_a2ctrainer_save_restore
7.87s call     trax/rl/actor_critic_joint_test.py::ActorCriticJointTest::test_awrjoint_save_restore
7.60s call     trax/rl/actor_critic_joint_test.py::ActorCriticJointTest::test_jointppotrainer_cartpole
7.35s call     trax/rl/actor_critic_joint_test.py::ActorCriticJointTest::test_jointa2ctrainer_cartpole
6.35s call     trax/rl/actor_critic_test.py::ActorCriticTest::test_sanity_a2ctrainer_cartpole
5.21s call     trax/rl/actor_critic_joint_test.py::ActorCriticJointTest::test_jointawrtrainer_cartpole
0.01s call     trax/rl/task_test.py::TaskTest::test_task_save_init
0.01s call     trax/rl/task_test.py::TaskTest::test_trajectory_stream_final_state

(0.00 durations hidden.  Use -vv to show these durations.)
[33m[1m=================== 26 passed, 1 warnings in 313.19 seconds ====================[0m
set_status

# Catch-all for futureproofing.

pytest --disable-warnings \
  --ignore=trax/fastmath \
  --ignore=trax/layers \
  --ignore=trax/models \
  --ignore=trax/optimizers \
  --ignore=trax/rl \
  --ignore=trax/supervised \
  --ignore=trax/tf_numpy
[1m============================= test session starts ==============================[0m
platform linux -- Python 3.6.7, pytest-4.3.1, py-1.7.0, pluggy-0.8.0
rootdir: /home/travis/build/aryaman4/trax, inifile: pytest.ini
plugins: typeguard-2.9.1
[1mcollecting ... [0m[1m
collecting 1 item                                                              [0m[1m
collected 56 items                                                             [0m

trax/import_test.py [32m.[0m[36m                                                    [  1%][0m
trax/shapes_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[36m                                               [ 12%][0m
trax/trax2keras_test.py [32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[33ms[0m[33ms[0m[33ms[0m[33ms[0m[33ms[0m[33ms[0m[33ms[0m[33ms[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[32m.[0m[33ms[0m[33ms[0m[33ms[0m[33ms[0m[33ms[0m[33ms[0m[33ms[0m[33ms[0m[36m [ 98%]
[0m[33ms[0m[36m                                                                        [100%][0m

============================ slowest test durations ============================
132.18s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_2_False_False_True
125.97s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_2_True_True_True
125.53s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_2_True_False_True
124.26s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_2_False_True_True
31.04s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_2_True_False_False
31.02s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_2_False_True_False
28.95s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_2_True_True_False
28.83s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_2_False_False_False
10.72s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_2_False_False_True
5.68s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_1_1_2_False_False_True
2.77s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_1_1_2_False_True_True
2.65s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_1_1_2_True_True_True
2.59s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_1_1_2_True_False_True
1.44s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_2_True_True_True
1.23s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_2_True_False_True
1.21s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_2_False_True_True
1.09s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_None_False_False_True
1.08s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_None_False_True_True
1.06s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_None_True_False_True
1.05s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_None_True_True_True
0.64s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_2_True_False_True
0.62s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_2_True_True_True
0.56s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_2_False_True_True
0.55s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_2_False_False_True
0.15s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_1_1_2_True_True_False
0.14s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_1_1_2_True_False_False
0.14s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_1_1_2_False_False_False
0.13s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_1_1_2_False_True_False
0.13s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_2_False_False_False
0.04s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_2_True_False_False
0.04s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_None_False_False_False
0.04s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_2_True_True_False
0.04s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_None_True_False_False
0.04s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_None_True_True_False
0.04s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_2_False_True_False
0.04s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_None_False_True_False
0.03s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_None_False_False_False
0.01s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_None_True_True_False
0.01s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_1_1_2_True_True_True
0.01s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_1_1_2_False_False_True
0.01s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_None_True_False_False
0.01s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_1_1_2_True_False_True
0.01s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_None_False_True_True
0.01s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_None_True_False_True
0.01s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_1_1_2_False_True_True
0.01s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_None_False_True_False
0.01s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_None_False_False_True
0.01s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_2_True_True_True
0.01s call     trax/trax2keras_test.py::Trax2KerasTest::testTrain_2_1_None_True_True_True
0.01s setup    trax/trax2keras_test.py::Trax2KerasTest::testTrain_1_1_2_False_True_False
0.01s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_2_True_False_True
0.01s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_2_False_True_True
0.01s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_None_False_False_True
0.01s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_None_False_True_True
0.01s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_2_False_False_True
0.01s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_None_True_False_True
0.01s teardown trax/trax2keras_test.py::Trax2KerasTest::testTrain_0_1_None_True_True_True

(0.00 durations hidden.  Use -vv to show these durations.)
[33m[1m=========== 39 passed, 17 skipped, 22856 warnings in 671.33 seconds ============[0m
set_status

# TODO(traxers): Test tf-numpy separately.

# Check notebooks.

# TODO(afrozm): Add more.
jupyter nbconvert --ExecutePreprocessor.kernel_name=python3 \
  --ExecutePreprocessor.timeout=600 --to notebook --execute \
  trax/intro.ipynb;
[NbConvertApp] Converting notebook trax/intro.ipynb to notebook
[NbConvertApp] Executing notebook with kernel: python3
[NbConvertApp] Writing 11794 bytes to trax/intro.nbconvert.ipynb
set_status

# Print coverage report.
coverage report -m
Name                                           Stmts   Miss  Cover   Missing
----------------------------------------------------------------------------
trax/__init__.py                                   8      0   100%
trax/fastmath/__init__.py                         12      0   100%
trax/fastmath/jax.py                             158    106    33%   51, 60-65, 69, 74, 79-84, 88, 116-132, 148-152, 182, 187, 192-195, 213-224, 237-252, 270, 281, 284, 293-294, 320-335, 352-355, 374, 380, 385-399, 403-405
trax/fastmath/numpy.py                            10      6    40%   23-28
trax/fastmath/ops.py                             107     41    62%   68, 87, 92, 97, 102, 107, 112, 117, 122, 127, 132, 137, 142, 151-153, 159, 164, 169-184, 189, 194, 199, 204, 214, 219, 233
trax/fastmath/tf.py                               42     11    74%   45, 52, 59, 83, 91, 96, 100-102, 106-107
trax/jaxboard.py                                 181    148    18%   57-67, 82-88, 91-97, 101-104, 109-112, 115-117, 127-134, 144-162, 179-193, 204-221, 234-261, 272-300, 311-336, 344-364
trax/layers/__init__.py                           44      0   100%
trax/layers/acceleration.py                      105     83    21%   56-59, 65, 72-92, 96-98, 102-103, 107-109, 114-116, 120-121, 126-128, 132-133, 138-155, 160-165, 170-173, 178-187, 192-199, 224-239
trax/layers/activation_fns.py                     38     17    55%   56, 71, 88, 105, 114, 125-127, 137, 146, 160, 173, 182, 190-191, 202-203
trax/layers/attention.py                         198     94    53%   55, 73, 100-103, 114-151, 185, 191, 213, 268-269, 276, 285, 290-291, 307, 315-321, 326-332, 343, 347, 359-389, 402-403, 408-418, 423-441
trax/layers/base.py                              323    151    53%   265, 278-281, 300-313, 321, 357, 363, 382, 388, 396-397, 404, 452-454, 464-465, 470, 473-476, 481, 508-512, 519-545, 551-573, 645, 647, 649, 666-671, 676-683, 688-698, 704-713, 731-734, 740, 743, 749, 764-766, 768, 776-787, 792-809, 823-832, 837-842
trax/layers/combinators.py                       346    138    60%   78, 82, 126, 131, 190, 194, 198, 202, 213, 231, 241, 244, 267, 276-279, 282, 289-291, 294, 329-335, 340, 343-366, 369-389, 420, 490, 495, 530-560, 566, 576, 581, 592, 599-600, 605, 610-613, 616-618, 621-628, 644-648, 653, 656-662, 665-667, 709-718, 723, 731, 737, 749-751, 761, 772
trax/layers/convolution.py                        65     47    28%   36-48, 52-53, 56-67, 71-72, 77-88, 102, 112-121, 127
trax/layers/core.py                              128     44    66%   84, 89-90, 111, 205, 209, 212, 214, 219, 240, 248, 276, 282, 299, 332-339, 350-357, 368-378, 389-391
trax/layers/initializers.py                       99     51    48%   38-45, 52-61, 85, 87, 97, 99, 104-106, 108-109, 114, 121, 131, 136, 141, 147, 153-185, 190-192
trax/layers/metrics.py                            56     24    57%   65-78, 101, 118-121, 130-131, 138, 144-146, 154-161, 168-172
trax/layers/normalization.py                     112     81    28%   28-34, 38-65, 69-86, 89-93, 96-97, 100-102, 106-113, 149-165, 169-182, 186-196
trax/layers/pooling.py                            18     12    33%   50-54, 83-87, 116-120
trax/layers/research/__init__.py                   0      0   100%
trax/layers/research/efficient_attention.py      704    661     6%   49-51, 55-57, 71-80, 86-96, 128-197, 202-209, 214-225, 230-242, 304-326, 329-361, 368, 418-470, 474-541, 546, 551-556, 620-854, 908-938, 946-947, 950-972, 976-1024, 1029-1081, 1104-1120, 1123-1139, 1145-1196, 1200-1277, 1282-1386, 1402-1418, 1426-1427, 1430-1437, 1441-1471
trax/layers/research/position_encodings.py       241    210    13%   36-49, 52-86, 89-107, 116-122, 125-156, 159-162, 180-189, 205-214, 244-254, 267-282, 296-335, 338-361, 364-377, 391-393, 404-414, 417-440, 443-445
trax/layers/reversible.py                         67     48    28%   58-69, 73, 76-79, 86, 89-90, 93-95, 102-109, 114-126, 130-150
trax/layers/rnn.py                                88     63    28%   44-48, 51-65, 69-78, 83-87, 92-93, 113-117, 120-133, 137-145, 150-151, 173-177, 220-241, 249-252, 278-279, 297
trax/models/__init__.py                           45      0   100%
trax/models/atari_cnn.py                          20     12    40%   26-29, 39, 44-49, 66-71, 92-94, 102
trax/models/mlp.py                                13      1    92%   82
trax/models/neural_gpu.py                         21     14    33%   25, 31-43, 49-52, 71-74
trax/models/reformer/__init__.py                   0      0   100%
trax/models/reformer/reformer.py                 222    198    11%   38-40, 53-61, 90-107, 114-116, 121-123, 126-140, 147-214, 218-234, 259-273, 322-365, 435-472, 527-546, 570-592, 646-699, 791-865
trax/models/research/__init__.py                   0      0   100%
trax/models/research/bert.py                      65     50    23%   32-33, 36, 40, 51, 75-126, 133-139, 142-201
trax/models/research/skipping_transformer.py      72     57    21%   33-43, 47-48, 53-60, 63-123, 155-161
trax/models/resnet.py                             28     20    29%   26-42, 52-64, 90-101, 124, 135-138, 162
trax/models/rl.py                                 40     33    18%   25-29, 41-52, 71-102, 121-131
trax/models/rnn.py                                24     18    25%   47-61, 91, 135-160
trax/models/transformer.py                       106     80    25%   57-68, 116-129, 224-266, 326-358, 419-428, 500-512, 564-586, 592-606, 612-633
trax/optimizers/__init__.py                       17      0   100%
trax/optimizers/adafactor.py                      67     58    13%   60-65, 78-79, 82-94, 97-147
trax/optimizers/adam.py                           23     16    30%   56, 66-68, 71-83
trax/optimizers/base.py                           60     41    32%   55-63, 94, 98, 112-114, 136-153, 157-161, 165-176, 186, 189-192, 200-201, 206-208
trax/optimizers/momentum.py                       22     15    32%   48-53, 56, 59-73
trax/optimizers/rms_prop.py                       16      9    44%   32, 40, 43-50
trax/optimizers/sm3.py                            51     40    22%   36, 42-43, 46-54, 59-60, 63-66, 70-88, 91-98
trax/rl/__init__.py                               17     17     0%   16-47
trax/rl/actor_critic.py                          349    349     0%   17-814
trax/rl/actor_critic_joint.py                    230    230     0%   17-654
trax/rl/advantages.py                             41     41     0%   17-160
trax/rl/distributions.py                          60     60     0%   17-185
trax/rl/normalization.py                          58     58     0%   17-113
trax/rl/rl_layers.py                              86     86     0%   17-223
trax/rl/serialization_utils.py                   103    103     0%   17-370
trax/rl/space_serializer.py                       86     86     0%   16-217
trax/rl/task.py                                  283    283     0%   17-713
trax/rl/training.py                              186    186     0%   17-424
trax/rl_trainer.py                                86     86     0%   16-230
trax/shapes.py                                    42      9    79%   45, 47, 53, 58, 68, 72, 98, 131, 138
trax/supervised/__init__.py                       11      0   100%
trax/supervised/decoding.py                       29     29     0%   16-78
trax/supervised/history.py                        27     14    48%   50, 54-56, 60-63, 68, 72-75, 78
trax/supervised/inputs.py                        247    211    15%   67-85, 88, 91, 94, 99, 104, 109, 114, 119, 136-157, 173-206, 221-246, 251-262, 292-345, 367-398, 422-437, 459-470, 493-507, 512-515, 538-559, 567, 572-574, 579-582, 603-633
trax/supervised/lr_schedules.py                   63     47    25%   36, 48, 54, 85-110, 140-151, 155-160, 176, 196-197, 207-212
trax/supervised/tf_inputs.py                     383    321    16%   44-45, 50-51, 79-95, 101-110, 117-134, 144-168, 196-225, 230-231, 235-238, 244-264, 270-277, 290-293, 301-312, 320-343, 349-361, 367-375, 383-395, 404-418, 425-437, 443-457, 464-468, 474-490, 500-530, 541-563, 583-598, 617-628, 634-644, 651-660, 700-738, 762-766, 779-800
trax/supervised/trainer_lib.py                   467    392    16%   88-164, 168, 172, 177-181, 185-190, 195-199, 203-207, 211, 217-218, 232-276, 283-311, 316-331, 335-352, 369-381, 385-394, 398-399, 403-424, 428-441, 444, 448-457, 461-468, 484-504, 507, 511-520, 523, 528, 531-536, 586-659, 665, 671-716, 722-726, 732-752, 756-759, 773-792, 812-814, 830-841, 847-859, 865-870, 875, 880-886, 891-896, 901-903
trax/supervised/training.py                      243    189    22%   105-145, 156-193, 198, 203, 208, 212-215, 229-239, 259-273, 284-303, 316-325, 329, 339-355, 366-381, 396-403, 424-429, 433, 437, 441, 445, 449, 453, 457-460, 488-494, 498, 502, 506, 510, 514, 518, 521, 524-525, 532-533, 538-540, 544-547, 553-561, 566-572
trax/test_utils.py                                10      5    50%   30-36
trax/tf_numpy/__init__.py                          0      0   100%
trax/tf_numpy/extensions/__init__.py               5      0   100%
trax/tf_numpy/extensions/extensions.py           410    258    37%   44-48, 53, 83, 100, 109, 120, 126, 151-171, 196-215, 238-264, 274-279, 334-339, 345-351, 384, 392, 397, 438, 443, 474-489, 519-520, 549-550, 570-604, 781, 786-787, 805-832, 849-851, 873-877, 880, 884, 888, 892-895, 911, 914, 917, 920, 929-937, 951-962, 985-995, 1016-1053, 1074-1155, 1159-1163, 1176, 1189, 1193, 1200-1209
trax/tf_numpy/numpy/__init__.py                   20      1    95%   28
trax/tf_numpy/numpy_impl/__init__.py               4      0   100%
trax/tf_numpy/numpy_impl/array_ops.py            721    468    35%   45, 61, 80, 105, 126, 143-149, 154-180, 195, 216-219, 228, 230, 232, 282, 294-296, 313, 318, 341-358, 364-381, 388-416, 421-446, 463-464, 489-490, 511-512, 535-560, 566, 570-577, 582-593, 598-609, 624-627, 660, 666, 675-676, 681-687, 695, 713, 719, 725-726, 731-733, 738-745, 751-758, 764-767, 785-788, 793-817, 822-836, 847, 851, 853, 856, 866, 885-886, 903-904, 946-997, 1034-1051, 1095, 1108, 1111, 1117-1118, 1123-1126, 1136, 1140, 1145-1157, 1174-1175, 1179, 1193-1209, 1214-1218, 1224, 1235, 1240-1244, 1249-1255, 1261-1266, 1271-1276, 1280-1284, 1300-1313, 1318, 1323, 1329-1338, 1343-1347, 1354-1361, 1366-1387, 1392-1405, 1411-1424, 1430-1437, 1442, 1447, 1452-1460, 1465-1480, 1485-1517, 1523-1549
trax/tf_numpy/numpy_impl/arrays.py               111     37    67%   34, 79, 81, 84, 88, 90, 95, 99, 130, 135, 139, 142-145, 158, 163, 166, 169, 172, 180-183, 197, 216, 219, 222, 234, 265-269, 272, 276, 280, 284-287
trax/tf_numpy/numpy_impl/dtypes.py                44      3    93%   74, 79, 98
trax/tf_numpy/numpy_impl/math_ops.py             653    428    34%   52-53, 61-62, 76-77, 85-88, 92-95, 99, 109-115, 122-124, 134, 139-144, 149-154, 159-167, 181-182, 188, 193-197, 202-254, 259, 264, 269, 274, 279-285, 290, 296-305, 310-312, 318-320, 328-330, 338-349, 354-365, 370, 374-393, 398, 403-409, 413-423, 428, 433, 438, 443-447, 473, 488, 493, 498, 503, 508, 513, 518, 523, 528-532, 537, 542, 547, 552, 557, 562, 567, 572, 577, 582, 587, 592, 597-599, 604, 612-621, 626-630, 635, 640-642, 647, 652-654, 659, 664, 669-670, 675, 680, 687-689, 703-711, 716, 721, 726, 731, 736, 741, 746, 751, 756-759, 764, 769-786, 820-821, 827, 832, 837, 842, 857-861, 865-867, 872, 877, 882, 887-888, 902-928, 933-939, 944, 950-956, 961-974, 979, 986-1005, 1010-1021, 1025-1031, 1036, 1041, 1046-1049, 1054-1104, 1109-1122, 1128-1144
trax/tf_numpy/numpy_impl/random.py                11      4    64%   38-40, 53
trax/tf_numpy/numpy_impl/utils.py                168     56    67%   34, 38-46, 58, 88, 93-95, 121, 135-137, 146, 157, 192-197, 269-274, 294, 301, 314, 316, 323, 328, 338, 343, 348-358, 367, 371-373, 378, 383-387, 392-396
trax/trainer.py                                   87     87     0%   16-162
trax/trainer_flags.py                             28     28     0%   17-87
trax/trax2keras.py                                60      8    87%   32-33, 35, 145, 147, 149, 160-163
----------------------------------------------------------------------------
TOTAL                                           9491   6878    28%

exit $STATUS
travis_time:end:0b4e31ca:start=1595218999020226275,finish=1595221014992169425,duration=2015971943150,event=script
[0K[32;1mThe command "./oss_scripts/oss_tests.sh" exited with 0.[0m
travis_fold:start:cache.2
[0Kstore build cache
travis_time:start:1fc25dba
[0Ktravis_time:end:1fc25dba:start=1595221014997930751,finish=1595221015000777392,duration=2846641,event=cache
[0Ktravis_time:start:0bc33958
[0K[32;1mchanges detected, packing new archive[0m
[32;1muploading master/cache--linux-xenial-a6812ccfb4dc0afee8dfe70bac496b870a786fe4512527e931715ec7de567dfe--python-3.6.tgz[0m
[32;1mcache uploaded[0m
travis_time:end:0bc33958:start=1595221015004253865,finish=1595221074157700487,duration=59153446622,event=cache
[0Ktravis_fold:end:cache.2
[0K
travis_fold:start:after_success
[0Ktravis_time:start:07aa8a90
[0K$ coveralls
Submitting coverage to coveralls.io...
Could not submit coverage: 422 Client Error: Unprocessable Entity for url: https://coveralls.io/api/v1/jobs
Traceback (most recent call last):
  File "/home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages/coveralls/api.py", line 205, in wear
    response.raise_for_status()
  File "/home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages/requests/models.py", line 941, in raise_for_status
    raise HTTPError(http_error_msg, response=self)
requests.exceptions.HTTPError: 422 Client Error: Unprocessable Entity for url: https://coveralls.io/api/v1/jobs

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages/coveralls/cli.py", line 85, in main
    result = coverallz.wear()
  File "/home/travis/virtualenv/python3.6.7/lib/python3.6/site-packages/coveralls/api.py", line 208, in wear
    raise CoverallsException('Could not submit coverage: {}'.format(e))
coveralls.exception.CoverallsException: Could not submit coverage: 422 Client Error: Unprocessable Entity for url: https://coveralls.io/api/v1/jobs
travis_time:end:07aa8a90:start=1595221074161729848,finish=1595221076343930471,duration=2182200623,event=after_success
[0Ktravis_fold:end:after_success
[0K
Done. Your build exited with 0.