chore: Make the name of the colab endpoint unique between BigQuery and Spanner. This avoids clashes if the same notebook uses visualizers for both BigQuery graphs and Spanner graphs, which can cause malfunctions in the visualizers.

Author: ericfe-google

bigquery_magics/bigquery.py

@@ -680,9 +680,11 @@ def _add_graph_widget(query_result):
     try:
         from google.colab import output
 
-        output.register_callback("graph_visualization.Query", _colab_query_callback)
         output.register_callback(
-            "graph_visualization.NodeExpansion", _colab_node_expansion_callback
+            "bigquery.graph_visualization.Query", _colab_query_callback
+        )
+        output.register_callback(
+            "bigquery.graph_visualization.NodeExpansion", _colab_node_expansion_callback
         )
 
         # In colab mode, the Javascript doesn't use the port value we pass in, as there is no
@@ -703,6 +705,13 @@ def _add_graph_widget(query_result):
         port=port,
         params=query_result.to_json().replace("\\", "\\\\").replace('"', '\\"'),
     )
+    html_content = html_content.replace(
+        '"graph_visualization.Query"', '"bigquery.graph_visualization.Query"'
+    )
+    html_content = html_content.replace(
+        '"graph_visualization.NodeExpansion"',
+        '"bigquery.graph_visualization.NodeExpansion"',
+    )
     IPython.display.display(IPython.core.display.HTML(html_content))