Merge pull request #195 from jax-ml:jax-pin

PiperOrigin-RevId: 548227743
jax-ml · Jul 14, 2023 · d98c427 · d98c427
2 parents ec5b82d + 765a353
commit d98c427
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,15 +12,15 @@ dependencies = [
 
 [project.optional-dependencies]
 cuda12 = [
-  "jaxlib@ https://storage.googleapis.com/jax-releases/nightly/cuda12/jaxlib-0.4.14.dev20230714+cuda12.cudnn89-cp310-cp310-manylinux2014_x86_64.whl"
+  "jaxlib @ https://storage.googleapis.com/jax-releases/nightly/cuda12/jaxlib-0.4.14.dev20230714+cuda12.cudnn89-cp310-cp310-manylinux2014_x86_64.whl"
 ]
 cuda12_pip = [
   "jaxlib[cuda12_pip] @ https://storage.googleapis.com/jax-releases/nightly/cuda12/jaxlib-0.4.14.dev20230714+cuda12.cudnn89-cp310-cp310-manylinux2014_x86_64.whl"
 ]
 cuda11 = [
   "jaxlib @ https://storage.googleapis.com/jax-releases/nightly/cuda118/jaxlib-0.4.14.dev20230714+cuda11.cudnn86-cp310-cp310-manylinux2014_x86_64.whl"
 ]
-cuda11 = [
+cuda11_pip = [
   "jaxlib[cuda11_pip] @ https://storage.googleapis.com/jax-releases/nightly/cuda118/jaxlib-0.4.14.dev20230714+cuda11.cudnn86-cp310-cp310-manylinux2014_x86_64.whl"
 ]
 tests = [