From feac939938e18e4b17550e62f59ac1e023ab2822 Mon Sep 17 00:00:00 2001 From: Jake VanderPlas Date: Thu, 1 Aug 2024 13:29:05 -0700 Subject: [PATCH] Remove experimental_array_api target This target is no longer needed since the array_api module is now built-in to jax.numpy. Followup to https://github.com/google/jax/pull/22818. PiperOrigin-RevId: 658518474 --- jax/BUILD | 9 --------- tests/BUILD | 2 +- 2 files changed, 1 insertion(+), 10 deletions(-) diff --git a/jax/BUILD b/jax/BUILD index 69be704ba57c..96e5987c1c1e 100644 --- a/jax/BUILD +++ b/jax/BUILD @@ -985,15 +985,6 @@ pytype_library( deps = [":jax"], ) -pytype_library( - name = "experimental_array_api", - visibility = [":internal"] + jax_visibility("array_api"), - deps = [ - ":experimental", - ":jax", - ], -) - pytype_library( name = "experimental_sparse", srcs = glob( diff --git a/tests/BUILD b/tests/BUILD index adea49cac293..d338b15af0de 100644 --- a/tests/BUILD +++ b/tests/BUILD @@ -59,7 +59,7 @@ py_test( srcs = ["array_api_test.py"], deps = [ "//jax", - "//jax:experimental_array_api", + "//jax:experimental", "//jax:test_util", ] + py_deps("absl/testing"), )