Skip to content

Commit

Permalink
[export] Clean up BUILD targets for jax.experimental.export
Browse files Browse the repository at this point in the history
jax.experimental.export is deprecated and will be removed in a future version of JAX.

See migration guide at: https://jax.readthedocs.io/en/latest/export/export.html#migration-guide-from-jax-experimental-export

PiperOrigin-RevId: 647562073
  • Loading branch information
gnecula authored and jax authors committed Jun 28, 2024
1 parent 016c1e8 commit 24b42ee
Show file tree
Hide file tree
Showing 4 changed files with 0 additions and 5 deletions.
1 change: 0 additions & 1 deletion jax/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -181,7 +181,6 @@ py_library(
] + jax_internal_export_back_compat_test_util_visibility,
deps = [
":jax",
"//jax/experimental/export",
] + py_deps("numpy"),
)

Expand Down
1 change: 0 additions & 1 deletion jax/experimental/jax2tf/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,5 @@ py_library(
visibility = jax_visibility("jax2tf_internal"),
deps = [
"//jax",
"//jax/experimental/export",
] + py_deps("numpy") + py_deps("tensorflow_core") + jax2tf_deps,
)
2 changes: 0 additions & 2 deletions tests/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -1465,7 +1465,6 @@ jax_test(
],
deps = [
"//jax:internal_test_harnesses",
"//jax/experimental/export",
],
)

Expand All @@ -1488,7 +1487,6 @@ jax_test(
],
deps = [
"//jax:internal_test_harnesses",
"//jax/experimental/export",
],
)

Expand Down
1 change: 0 additions & 1 deletion tests/pallas/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -357,6 +357,5 @@ jax_test(
"//jax:pallas",
"//jax:pallas_gpu", # build_cleaner: keep
"//jax:pallas_tpu", # build_cleaner: keep
"//jax/experimental/export",
],
)

0 comments on commit 24b42ee

Please sign in to comment.