chore: adopt new local wheel installation logic

vam-google · jax authors · commit a073476fa0b9 · 2024-06-10T11:41:52.000-07:00
PiperOrigin-RevId: 641972325
diff --git a/WORKSPACE b/WORKSPACE
@@ -15,6 +15,9 @@ python_init_repositories(
         "3.12": "//build:requirements_lock_3_12.txt",
         "3.13": "//build:requirements_lock_3_13.txt",
     },
+    local_wheel_workspaces = ["//jaxlib:jax.bzl"],
+    local_wheel_dist_folder = "../dist",
+    default_python_version = "system",
 )
 
 load("@xla//third_party/py:python_init_toolchains.bzl", "python_init_toolchains")

Original file line number	Diff line number	Diff line change
`@@ -15,6 +15,9 @@ python_init_repositories(`
`15`	`15`	`"3.12": "//build:requirements_lock_3_12.txt",`
`16`	`16`	`"3.13": "//build:requirements_lock_3_13.txt",`
`17`	`17`	`},`
	`18`	`+ local_wheel_workspaces = ["//jaxlib:jax.bzl"],`
	`19`	`+ local_wheel_dist_folder = "../dist",`
	`20`	`+ default_python_version = "system",`
`18`	`21`	`)`
`19`	`22`
`20`	`23`	`load("@xla//third_party/py:python_init_toolchains.bzl", "python_init_toolchains")`