Update various Python packaging configs

PiperOrigin-RevId: 480929772
google · Oct 13, 2022 · 8693a58 · 8693a58
1 parent 2ee3c1d
commit 8693a58
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/build_and_minimal_test.yml b/.github/workflows/build_and_minimal_test.yml
@@ -14,7 +14,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.7, 3.8, 3.9]
+        python-version: ['3.7', '3.8', '3.9', '3.10']
 
     steps:
     - uses: actions/checkout@v2

diff --git a/setup.py b/setup.py
@@ -36,7 +36,7 @@
     install_requires=[
         'absl-py',
         'dm-haiku',
-        'jax',
+        'jax>=0.3.19',
         'jaxlib',
         'msgpack',
         'optax',
@@ -52,6 +52,7 @@
         'Programming Language :: Python :: 3.7',
         'Programming Language :: Python :: 3.8',
         'Programming Language :: Python :: 3.9',
+        'Programming Language :: Python :: 3.10',
         'Topic :: Scientific/Engineering :: Artificial Intelligence',
     ],
 )