added windows constraint in noxfile

peng-lab · Oct 24, 2023 · 6b9c765 · 6b9c765
1 parent 84b5632
commit 6b9c765
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/noxfile.py b/noxfile.py
@@ -14,12 +14,13 @@ def tests(session: Session) -> None:
     """Run the test suite."""
     if platform.system() == "Windows":
         session.install(
-            "jax[cpu]",
+            "jax[cpu]==0.4.11",
             "-f",
             "https://whls.blob.core.windows.net/unstable/index.html",
             "--use-deprecated",
             "legacy-resolver",
         )
+        session.install("ml-dtypes==0.2.0")
     session.install(".")
     session.install(
         "dask",