Unable to import BigQuery data into GCP AI Notebook

Question

I had previously used the code below to import data from bigquery to an AI notebook instance in GCP. For unknown reasons it stopped working and gives me the following error: "ImportError: cannot import name 'bigquery_storage_v1beta1' from 'google.cloud' (unknown location)". It may have started after I began cloning my github repository (I delete the AI notebook instance to avoid a charge), but I can't be sure. Thoughts?

import pandas as pd

pd.options.display.max_rows = 200 
from google.cloud import bigquery
from google.cloud import storage

import numpy as np

query="""
SELECT *
FROM
  [table_name] """

df = bigquery.Client().query(query).to_dataframe()
df.head()

Answer 1

The solution was updating the packages with the following commands:

First:

!sudo /opt/conda/bin/conda install -c conda-forge google-cloud-bigquery google-cloud-bigquery-storage pandas pyarrow --yes

Then:

%pip install google-cloud-bigquery