[Bug]: root/.cache/bazel/_bazel_root/5fb069ff7e8788e0e7d803c89624f2bc/execroot/spulib/bazel-out/k8-fastbuild/bin/spu/tests/jnp_dynamic_ring_test.runfiles/spulib/spu/tests/jnp_dynamic_ring_test.py", line 15, in <module> import jax ModuleNotFoundError: No module named 'jax'

[zhangwaer]

Issue Type

Build/Install

Modules Involved

Others

Have you reproduced the bug with SPU HEAD?

Yes

Have you searched existing issues?

Yes

SPU Version

0.5.0.dev20240805

OS Platform and Distribution

linux20.04

Python Version

3.8

Compiler Version

GCC11.2.0

Current Behavior?

when i installed "https://github.com/secretflow/spu/tree/ditto_icml24/examples/python/ml/experimental_mp"
and i use the code "bazel run //spu/tests:jnp_dynamic_ring_test", it report as follows，hope to get your advice, thanks very much!!!:
root/.cache/bazel/_bazel_root/5fb069ff7e8788e0e7d803c89624f2bc/execroot/spulib/bazel-out/k8-fastbuild/bin/spu/tests/jnp_dynamic_ring_test.runfiles/spulib/spu/tests/jnp_dynamic_ring_test.py", line 15, in import jax ModuleNotFoundError: No module named 'jax'

Standalone code to reproduce the issue

root/.cache/bazel/_bazel_root/5fb069ff7e8788e0e7d803c89624f2bc/execroot/spulib/bazel-out/k8-fastbuild/bin/spu/tests/jnp_dynamic_ring_test.runfiles/spulib/spu/tests/jnp_dynamic_ring_test.py", line 15, in <module> import jax ModuleNotFoundError: No module named 'jax'

Relevant log output

No response

[anakinxc]

Please make sure you have jax installed

[zhangwaer]

Please make sure you have jax installed

Thanks very much!!!, i succeed through "bazel clean --expunge"