diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 640905c7..ed3bbdec 100644 --- 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/cancel-workflow-action@0.8.0 diff --git a/checkpoint/CHANGELOG.md b/checkpoint/CHANGELOG.md index 26707ff9..b28df19b 100644 --- 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 index a4f2696b..c663256b 100644 --- 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 index 90f05c1c..68eb7dd2 100644 --- 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='orbax-dev@google.com'}] 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 index bd208eb1..e2b56c67 100644 --- 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