Add 3.14 to version matrix

Author: patrick-kidger

.github/workflows/run_tests.yml

@@ -26,7 +26,7 @@ jobs:
   run-tests:
     strategy:
       matrix:
-        python-version: [ "3.10", "3.12" ]
+        python-version: [ "3.10", "3.12", "3.14.0-rc.3"]
         os: [ ubuntu-latest ]
       fail-fast: false
     runs-on: ${{ matrix.os }}

pyproject.toml

@@ -47,11 +47,11 @@ tests = [
     "equinox>=0.13.1",
     "ipython>=8.37.0",
     "jax>=0.5.3",
-    "mlx[cpu]>=0.29.1",
-    "numpy<2",
+    "mlx[cpu]>=0.29.1; python_version < '3.14'",
+    "numpy",
     "pytest>=8.4.2",
     "pytest-asyncio>=1.2.0",
-    "tensorflow>=2.18.1",
+    "tensorflow>=2.18.1; python_version < '3.14'",
     "typeguard==2.13.3",
 ]
 

test/test_tf_dtype.py

@@ -1,11 +1,11 @@
 # Tensorflow dependency kept in a separate file, so that we can optionally exclude it
 # more easily.
-import tensorflow as tf
-
 from jaxtyping import UInt
 
 
 def test_tf_dtype():
+    import tensorflow as tf
+
     x = tf.constant(1, dtype=tf.uint8)
     y = tf.constant(1, dtype=tf.float32)
     hint = UInt[tf.Tensor, "..."]