Jax error with the create_notebook.py code

[BismahGhafoor]

I am trying to make the jupyter notebook but the script is failing to open the .pkl files.

This is the error I am getting:

(base) [bg171@alice-login01 FinalProject]$ python3 test5.py I0716 10:35:35.385460 23456246728512 test5.py:126] now processing Q09472_and_P57682 Traceback (most recent call last): File "/lustre/alice3/scratch/alice/b/bg171/FinalProject/test5.py", line 159, in app.run(main) File "/home/b/bg171/miniconda3/lib/python3.12/site-packages/absl/app.py", line 308, in run _run_main(main, args) File "/home/b/bg171/miniconda3/lib/python3.12/site-packages/absl/app.py", line 254, in _run_main sys.exit(main(argv)) ^^^^^^^^^^ File "/lustre/alice3/scratch/alice/b/bg171/FinalProject/test5.py", line 139, in main pae_mtx, iptm_score = obtain_pae_and_iptm(result_subdir=result_subdir, ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/lustre/alice3/scratch/alice/b/bg171/FinalProject/test5.py", line 50, in obtain_pae_and_iptm data = pickle.load(f) ^^^^^^^^^^^^^^ ModuleNotFoundError: No module named 'jax._src.device_array'

I am using jax and jaxlib versions 0.4.21.

some guidance would be appreciated a lot.

[dingquanyu]

Hi @BismahGhafoor

It seems only jax.0.4.23 works with this import. Could you try installing jax0.4.23 ?

Yours Dingquan