chore: fix google3 import colabsqlviz tests#2499

GarrettWu merged 2 commits intomaingoogleapis/python-bigquery-dataframes:mainfrom

garrettwu-fixgoogleapis/python-bigquery-dataframes:garrettwu-fixCopy head branch name to clipboard