Drop python 3.6 (dropped by JAX) and add python 3.9 to tests.

PiperOrigin-RevId: 391440173
google · Aug 18, 2021 · 42cf4d5 · 42cf4d5
1 parent 87c0310
commit 42cf4d5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -1,8 +1,8 @@
 language: python
 python:
-  - "3.6"
   - "3.7"
   - "3.8"
+  - "3.9"
 install:
   - pip install --upgrade pip
   - pip install more-itertools numpy jaxlib tensorflow tensorflow-datasets --upgrade

diff --git a/setup.py b/setup.py
@@ -29,15 +29,11 @@
 
 
 INSTALL_REQUIRES = [
-    'jax>=0.1.77',
+    'jax>=0.2.18',
     'frozendict>=1.2',
 ]
 
 
-if sys.version_info < (3, 7):
-  INSTALL_REQUIRES += ['dataclasses>=0.7']
-
-
 def _get_version() -> str:
   """Returns the package version.
 
@@ -76,9 +72,8 @@ def _get_version() -> str:
     long_description=long_description,
     long_description_content_type='text/markdown',
     description='Fast and Easy Infinite Neural Networks in Python',
-    python_requires='>=3.6',
+    python_requires='>=3.7',
     classifiers=[
-        'Programming Language :: Python :: 3.6',
         'Programming Language :: Python :: 3.7',
         'Programming Language :: Python :: 3.8',
         'Programming Language :: Python :: 3.9',