biot5x/examples/finetunning_biot5x_blurb.ipynb has a problem as to typing_extensions.

[yayamamo]

After compliting the gsutil step and then during executing the biot5x/src/finetune_biot5x.py, I encountered the following error:

~/.local/lib/python3.8/site-packages/tensorflow/python/types/trace.py in <module>
     29 from typing import Any, List, Optional, Sequence
     30 
---> 31 from typing_extensions import Protocol
     32 from typing_extensions import runtime_checkable
     33 

ModuleNotFoundError: No module named 'typing_extensions'

I confirmed that pip install typing_extensions has been done successfully.

• Ubuntu 20.04.6 LTS (Focal Fossa)
• 5.15.133.1-microsoft-standard-WSL2
• Python 3.11.5

[heraclex12]

Did you try to run the biot5x/src/finetune_biot5x.py script in an environment? If not, you should try to create your own one and then install all the required packages including pip install typing-extensions.

[yayamamo]

Thanks. When trying the script at my PC, the following error occurred. I have already installed the jax module by pip install jax .

2023-12-15 16:51:48.224877: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
Traceback (most recent call last):
  File "biot5x/src/finetune_biot5x.py", line 38, in <module>
    from jax.experimental.global_device_array import GlobalDeviceArray
ModuleNotFoundError: No module named 'jax.experimental.global_device_array'