Importing bigquery_storage

I am trying the [blogs/babyweight/train_deploy.ipynb]( https://github.com/GoogleCloudPlatform/training-data-analyst/blob/master/blogs/babyweight/train_deploy.ipynb). Running the cells which preprocess and filter the data, I face an error-

```sh
cannot import name 'bigquery_storage_v1' from 'google.cloud' (unknown location)
```

This seems to me like a dependency that is missing specifically the `google-cloud-bigquery-storage` which is used here as I am able to use the other functions of `google.cloud`.