Update requirements and pypi workflow.

PiperOrigin-RevId: 802025431

Author: hbq1

.github/workflows/ci.yml

@@ -13,12 +13,12 @@ jobs:
 
     strategy:
       matrix:
-        python-version: ["3.10", "3.11", "3.12"]
+        python-version: ["3.11", "3.12", "3.13"]
         os: [ubuntu-latest]
 
     steps:
-      - uses: "actions/checkout@v4"
-      - uses: "actions/setup-python@v4"
+      - uses: "actions/checkout@v5"
+      - uses: "actions/setup-python@v5.3.0"
         with:
             python-version: "${{ matrix.python-version }}"
       - name: Run CI tests

.github/workflows/pypi-publish.yml

@@ -7,6 +7,8 @@ on:
 jobs:
   deploy:
     runs-on: ubuntu-latest
+    permissions:
+      id-token: write
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python
@@ -25,10 +27,8 @@ jobs:
         then
           echo "package ver. ($PACKAGE_VER) != release ver. ($RELEASE_VER)"; exit 1
         fi
-    - name: Build and publish
-      env:
-        TWINE_USERNAME: ${{ secrets.PYPI_USERNAME }}
-        TWINE_PASSWORD: ${{ secrets.PYPI_PASSWORD }}
+    - name: Build
       run: |
         python setup.py sdist bdist_wheel
-        twine upload dist/*
+    - name: Publish package distributions to PyPI
+      uses: pypa/gh-action-pypi-publish@release/v1

mctx/__init__.py

@@ -37,7 +37,7 @@
 from mctx._src.search import search
 from mctx._src.tree import Tree
 
-__version__ = "0.0.5"
+__version__ = "0.0.6"
 
 __all__ = (
     "ChanceRecurrentFnOutput",

requirements/requirements-test.txt

@@ -1,2 +1,2 @@
-absl-py>=0.9.0
-numpy>=1.18.0
+absl-py>=2.3.1
+numpy>=1.24.1

requirements/requirements.txt

@@ -1,3 +1,3 @@
-chex>=0.0.8
-jax>=0.4.25
-jaxlib>=0.4.25
+chex>=0.1.91
+jax>=0.7.0
+jaxlib>=0.7.0

requirements/requirements_examples.txt

@@ -1,2 +1,2 @@
-absl-py>=0.9.0
+absl-py>=2.3.1
 pygraphviz>=1.7

setup.py

@@ -58,7 +58,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.9',
+    python_requires='>=3.11',
     classifiers=[
         'Development Status :: 4 - Beta',
         'Intended Audience :: Science/Research',