Remove orbax-checkpoint support for Python 3.9.

PiperOrigin-RevId: 683192479
google · Oct 7, 2024 · d1b8aaa · d1b8aaa
1 parent 410ca56
commit d1b8aaa
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 8 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -18,11 +18,11 @@ jobs:
         working-directory: checkpoint
     strategy:
       matrix:
-        python-version: ["3.9", "3.10", "3.11"]
+        python-version: ["3.10", "3.11"]
         jax-version: ["newest"]
         include:
-          - python-version: "3.9"
-            jax-version: "0.4.26"  # keep in sync with minimum version in checkpoint/pyproject.toml
+          - python-version: "3.10"
+            jax-version: "0.4.34"  # keep in sync with minimum version in checkpoint/pyproject.toml
     steps:
     - name: Cancel previous
       uses: styfle/[email protected]

diff --git a/checkpoint/CHANGELOG.md b/checkpoint/CHANGELOG.md
@@ -7,6 +7,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.7.0] - 2024-10-07
+
+### Removed
+- Support for Python 3.9.
+
 ### Changed
 - Modernize type annotations using `from __future__ import annotations`.
 - Adjust `CompositeCheckpointHandler` behavior for unregistered items and empty

diff --git a/checkpoint/orbax/checkpoint/version.py b/checkpoint/orbax/checkpoint/version.py
@@ -16,7 +16,7 @@
 
 # A new PyPI release will be pushed everytime `__version__` is increased.
 # Also modify version and date in CHANGELOG.
-__version__ = '0.6.4'
+__version__ = '0.7.0'
 
 
 # TODO: b/362813406 - Add latest change timestamp and commit number.

diff --git a/checkpoint/pyproject.toml b/checkpoint/pyproject.toml
@@ -7,7 +7,7 @@ name = "orbax-checkpoint"
 description = "Orbax Checkpoint"
 readme = 'README.md'
 license = {file = 'LICENSE'}
-requires-python = '>=3.9'
+requires-python = '>=3.10'
 authors = [{name = 'Orbax Authors', email='[email protected]'}]
 classifiers=[
     'Development Status :: 4 - Beta',
@@ -26,8 +26,7 @@ dependencies = [
     'etils[epath,epy]',
     'typing_extensions',
     'msgpack',
-    'jax >= 0.4.26',
-    'jaxlib',
+    'jax >= 0.4.34',
     'numpy',
     'pyyaml',
     'tensorstore >= 0.1.60',

diff --git a/readthedocs.yml b/readthedocs.yml
@@ -6,7 +6,7 @@ version: 2
 build:
   os: ubuntu-22.04
   tools:
-    python: "3.9"
+    python: "3.10"
 
 sphinx:
   builder: html