diff --git a/.github/workflows/publish-to-pypi.yml b/.github/workflows/publish-to-pypi.yml index cdda3e8..54db03c 100644 --- a/.github/workflows/publish-to-pypi.yml +++ b/.github/workflows/publish-to-pypi.yml @@ -17,7 +17,7 @@ jobs: - name: Set up Python uses: actions/setup-python@v3 with: - python-version: '3.11' + python-version: '3.12' - name: Install dependencies run: | python -m pip install --upgrade pip diff --git a/.github/workflows/run-tests.yml b/.github/workflows/run-tests.yml index f304bbd..0a04d5f 100644 --- a/.github/workflows/run-tests.yml +++ b/.github/workflows/run-tests.yml @@ -16,7 +16,7 @@ jobs: strategy: fail-fast: false matrix: - python-version: ["3.10", "3.11"] + python-version: ["3.11", "3.12"] steps: - uses: actions/checkout@v4 diff --git a/README.md b/README.md index 53ed076..daae985 100644 --- a/README.md +++ b/README.md @@ -52,6 +52,6 @@ For the Jax example to work, you will additionally need to install `absl-py` and For additional examples see the `tests` directory. ## Notes -* `coremltools` supports up to python 3.11. Do not run hatch with a newer version. - Can be controlled using fx `export HATCH_PYTHON=python3.11` +* `coremltools` supports up to python 3.12. Do not run hatch with a newer version. + Can be controlled using fx `export HATCH_PYTHON=python3.12` * Run tests using `hatch run test:pytest tests` diff --git a/pyproject.toml b/pyproject.toml index d670633..dba7213 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -24,7 +24,7 @@ keywords=[ "stablehlo", "hlo", "xla", "coreml", "machinelearning", "ml", "coreml requires-python = ">=3.9" dependencies = [ - 'coremltools>=8.0; python_version >= "3.9" and python_version <= "3.11"', + 'coremltools>=8.1; python_version >= "3.10" and python_version <= "3.12"', "numpy~=2.0", # Jax is not actually a strict requirement for the main library. @@ -42,7 +42,7 @@ path = "stablehlo_coreml/__init__.py" packages = ["stablehlo_coreml"] [[tool.hatch.envs.test.matrix]] -python = ["3.10", "3.11"] +python = ["3.11", "3.12"] [tool.hatch.envs.test] randomize = true @@ -59,7 +59,7 @@ extra-dependencies = [ ] [[tool.hatch.envs.test-pytorch.matrix]] -python = ["3.10", "3.11"] +python = ["3.11", "3.12"] [tool.hatch.envs.test-pytorch] randomize = true