From e51e70e1b9db374b467b67cbd4734cc75883239e Mon Sep 17 00:00:00 2001
From: Fabian Pedregosa <pedregosa@google.com>
Date: Tue, 5 Dec 2023 10:22:19 +0100
Subject: [PATCH] Run pytype on Python 3.11

---
 test.sh | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/test.sh b/test.sh
index e8e53f78..55341116 100755
--- a/test.sh
+++ b/test.sh
@@ -64,13 +64,8 @@ pip wheel --verbose --no-deps --no-clean dist/optax*.tar.gz
 pip install optax*.whl
 
 # Check types with pytype.
-# Note: pytype does not support 3.11 as of 25.06.23
-# See https://github.com/google/pytype/issues/1308
-if [ `python -c 'import sys; print(sys.version_info.minor)'` -lt 11 ];
-then
-  pip install pytype
-  pytype `find optax/_src/ examples -name '*.py' | xargs` -k -d import-error
-fi;
+pip install pytype
+pytype `find optax/_src/ examples -name '*.py' | xargs` -k -d import-error
 
 # Run tests using pytest.
 # Change directory to avoid importing the package from repo root.