From cf732f25d860651b856acec5eca39819119a45b2 Mon Sep 17 00:00:00 2001 From: Sait Cakmak Date: Wed, 22 Jan 2025 14:40:02 -0500 Subject: [PATCH 1/3] Unpin jaxtyping --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index f3313a47..ed0d2f57 100644 --- a/setup.py +++ b/setup.py @@ -39,7 +39,7 @@ # Other requirements install_requires += [ "scipy", - "jaxtyping==0.2.19", + "jaxtyping", "mpmath>=0.19,<=1.3", # avoid incompatibiltiy with torch+sympy with mpmath 1.4 ] From 62236877e17ee5b3fa87ad163b882916a01532cf Mon Sep 17 00:00:00 2001 From: Sait Cakmak Date: Fri, 24 Jan 2025 22:06:23 -0500 Subject: [PATCH 2/3] Update CI --- .github/workflows/run_type_checked_test_suite.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/run_type_checked_test_suite.yml b/.github/workflows/run_type_checked_test_suite.yml index 5ec172dc..f21cdb27 100644 --- a/.github/workflows/run_type_checked_test_suite.yml +++ b/.github/workflows/run_type_checked_test_suite.yml @@ -21,7 +21,7 @@ jobs: - name: Set up Python uses: actions/setup-python@v2 with: - python-version: "3.8" + python-version: "3.10" - name: Install dependencies run: | if [[ ${{ matrix.pytorch-version }} = "latest" ]]; then @@ -32,4 +32,4 @@ jobs: pip install -e ".[test]" - name: Run unit tests run: | - pytest ${{ inputs.files_to_test }} --jaxtyping-packages=linear_operator,typeguard.typechecked + pytest ${{ inputs.files_to_test }} --jaxtyping-packages=typeguard.typechecked From d3dbd0aee4e0c554225bb618a91f702e43486353 Mon Sep 17 00:00:00 2001 From: Sait Cakmak Date: Fri, 24 Jan 2025 22:12:07 -0500 Subject: [PATCH 3/3] Update run_type_checked_test_suite.yml --- .github/workflows/run_type_checked_test_suite.yml | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/.github/workflows/run_type_checked_test_suite.yml b/.github/workflows/run_type_checked_test_suite.yml index f21cdb27..0039c74a 100644 --- a/.github/workflows/run_type_checked_test_suite.yml +++ b/.github/workflows/run_type_checked_test_suite.yml @@ -25,9 +25,10 @@ jobs: - name: Install dependencies run: | if [[ ${{ matrix.pytorch-version }} = "latest" ]]; then - pip install --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html; + pip install --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html else - pip install torch==1.11+cpu -f https://download.pytorch.org/whl/torch_stable.html; + pip install torch==1.11+cpu -f https://download.pytorch.org/whl/torch_stable.html + pip install "numpy<2" fi pip install -e ".[test]" - name: Run unit tests