mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-13 02:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove internal_test_util folder and packages from jax wheel.

Browse Source 
PiperOrigin-RevId: 736861450
This commit is contained in:
jax authors 2025-03-14 07:51:21 -07:00
parent 8ab33669e2
commit 8fbe3b1333
2 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
BUILD.bazel
View File

@ -31,9 +31,6 @@ transitive_py_deps(
"//jax:experimental",
"//jax:experimental_colocated_python",
"//jax:experimental_sparse",
"//jax:internal_export_back_compat_test_util",
"//jax:internal_test_harnesses",
"//jax:internal_test_util",
"//jax:lax_reference",
"//jax:pallas_experimental_gpu_ops",
"//jax:pallas_gpu_ops",

2
setup.py
View File

@ -50,7 +50,7 @@ setup(
long_description_content_type='text/markdown',
author='JAX team',
author_email='jax-dev@google.com',
packages=find_packages(exclude=["examples", "jax/src/internal_test_util"]),
packages=find_packages(exclude=["*examples*", "*internal_test_util*"]),
package_data={'jax': ['py.typed', "*.pyi", "**/*.pyi"]},
python_requires='>=3.10',
install_requires=[