[Bug] Fail to run train_deepscaler_nb.py

**Expected Behavior**

Successfully run `python3 examples/deepscaler/train_deepscaler_nb.py`.

**Actual Behavior**

**Steps to Reproduce the Problem**

1. `python3 examples/deepscaler/train_deepscaler_nb.py`

**Environment**

- **OS:** [e.g., Ubuntu, etc.] Ubuntu-22.04, tpu-v6e-4.
- **Project Version:** [e.g., 0.0.1] main-096080ffbb6529353b8e0897142bc425ed57fb3b

**Checklist**

- [x] I have searched the existing issues for a similar bug report.
- [x] I have provided all the required information in the "Environment" section.
- [x] I have provided a minimal, reproducible example.

**Would you like to help us fix it?**

Maybe I need a correct google package to run.