language: python
python:
  - "2.7"
  - "3.5"
  - "3.6"
  - "3.7"
install:
  - pip install --upgrade pip
  - pip install tensorflow tensorflow-datasets
  - git clone https://github.com/google/neural-tangents
  - pip install -e neural-tangents
script:
  - for f in neural-tangents/neural_tangents/tests/*.py; do python $f; done
language: python
python:
  - "3.5"
  - "3.6"
  - "3.7"
install:
  - pip install --upgrade pip
  - pip install tensorflow tensorflow-datasets
  - git clone https://github.com/google/neural-tangents
  - pip install -e neural-tangents
script:
  - for f in neural-tangents/neural_tangents/tests/*.py; do python $f; done

Remove python 2 from Travis CI.
Remove python 2 from Travis CI.

PiperOrigin-RevId: 290138028
