Migrate dependencies on jax experimental targets to jax/experimental/BUILD.

danielsuo · tensorflower-gardener · commit 94ee88feb958 · 2025-08-01T06:54:56.000-07:00
PiperOrigin-RevId: 789746372
diff --git a/tensorflow_probability/python/experimental/fastgp/BUILD b/tensorflow_probability/python/experimental/fastgp/BUILD
@@ -176,7 +176,7 @@ py_library(
     srcs = ["linalg.py"],
     deps = [
         ":partial_lanczos",
-        # jax:experimental_sparse dep,
+        # jax/experimental:sparse dep,
         "//tensorflow_probability/python/internal/backend/jax",
     ],
 )

Original file line number	Diff line number	Diff line change
`@@ -176,7 +176,7 @@ py_library(`
`176`	`176`	`srcs = ["linalg.py"],`
`177`	`177`	`deps = [`
`178`	`178`	`":partial_lanczos",`
`179`		`- # jax:experimental_sparse dep,`
	`179`	`+ # jax/experimental:sparse dep,`
`180`	`180`	`"//tensorflow_probability/python/internal/backend/jax",`
`181`	`181`	`],`
`182`	`182`	`)`