Merge pull request #105 from cornellius-gp/jaxtyping-1

Unpin jaxtyping

Author: Balandat

.github/workflows/run_type_checked_test_suite.yml

@@ -21,15 +21,16 @@ 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
-          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
       run: |
-        pytest ${{ inputs.files_to_test }} --jaxtyping-packages=linear_operator,typeguard.typechecked
+        pytest ${{ inputs.files_to_test }} --jaxtyping-packages=typeguard.typechecked

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
 ]