separate dev from prod requirements, relax prod requirements

PiperOrigin-RevId: 550911347
google · Jul 25, 2023 · 5123c60 · 5123c60
1 parent 59965be
commit 5123c60
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 13 deletions.
diff --git a/WORKSPACE b/WORKSPACE
@@ -60,7 +60,7 @@ load("@rules_python//python:pip.bzl", "pip_parse")
 pip_parse(
     name = "jaxite_deps",
     python_interpreter_target = interpreter,
-    requirements_lock = "//:requirements.txt",
+    requirements_lock = "//:requirements_dev.txt",
 )
 
 load("@jaxite_deps//:requirements.bzl", "install_deps")

diff --git a/pyproject.toml b/pyproject.toml
@@ -28,9 +28,8 @@ classifiers = [
   "Programming Language :: Python :: Implementation :: PyPy",
 ]
 dependencies = [
-  "jax==0.4.13",
-  "jaxlib==0.4.13",
-  "numpy==1.25.1",
+  "jax~=0.4.13",
+  "jaxlib~=0.4.13",
 ]
 
 [project.urls]

diff --git a/requirements.txt b/requirements.txt
@@ -1,9 +1,2 @@
-attrs==23.1.0
-hypothesis==6.79.1
-jax==0.4.13
-jaxlib==0.4.13
-ml-dtypes==0.2.0
-numpy==1.25.1
-opt-einsum==3.3.0
-scipy==1.10.1
-sortedcontainers==2.4.0
+jax~=0.4.13
+jaxlib~=0.4.13
diff --git a/requirements_dev.txt b/requirements_dev.txt
@@ -0,0 +1,9 @@
+attrs==23.1.0
+hypothesis==6.79.1
+jax==0.4.13
+jaxlib==0.4.13
+ml-dtypes==0.2.0
+numpy==1.25.1
+opt-einsum==3.3.0
+scipy==1.10.1
+sortedcontainers==2.4.0