From 46fbf8a9e62f2af55247e64ae87b1d0523e94ed2 Mon Sep 17 00:00:00 2001
From: Ross Hemsley <rhemsley@google.com>
Date: Thu, 18 Nov 2021 11:26:06 -0800
Subject: [PATCH] Drop Python 3.6 support, which is no longer supported by JAX.
 (change 2 of 2)

Fixes https://github.com/deepmind/optax/issues/222

PiperOrigin-RevId: 410856130
---
 .github/workflows/tests.yml | 2 +-
 optax/__init__.py           | 2 +-
 setup.py                    | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 5d63ce603..972fa45bd 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -13,7 +13,7 @@ jobs:
 
     strategy:
       matrix:
-        python-version: ["3.6", "3.7", "3.8", "3.9"]
+        python-version: ["3.7", "3.8", "3.9"]
         os: [ubuntu-latest]
 
     steps:
diff --git a/optax/__init__.py b/optax/__init__.py
index 9fdfc7f1c..e03c62957 100644
--- a/optax/__init__.py
+++ b/optax/__init__.py
@@ -148,7 +148,7 @@
 from optax._src.wrappers import MultiSteps
 from optax._src.wrappers import MultiStepsState
 
-__version__ = "0.0.91"
+__version__ = "0.1.0"
 
 __all__ = (
     "adabelief",
diff --git a/setup.py b/setup.py
index 8f39cfd8c..9372530ef 100644
--- a/setup.py
+++ b/setup.py
@@ -59,7 +59,7 @@ def _parse_requirements(path):
     tests_require=_parse_requirements(
         os.path.join(_CURRENT_DIR, 'requirements', 'requirements-test.txt')),
     zip_safe=False,  # Required for full installation.
-    python_requires='>=3.6',
+    python_requires='>=3.7',
     classifiers=[
         'Development Status :: 5 - Production/Stable',
         'Environment :: Console',