Fix notebook template to be valid ipynb.

DataBiosphere · Jul 28, 2023 · db49cc8 · db49cc8
1 parent be2e94f
commit db49cc8
Showing 1 changed file with 36 additions and 15 deletions.
diff --git a/service/src/main/resources/export/notebook_template.ipynb b/service/src/main/resources/export/notebook_template.ipynb
@@ -1,15 +1,36 @@
-import pandas
-import os
-
-# This query was generated for dataset ${underlayName} in study ${studyName} at ${timestamp}.
-${entityName}_sql = """
-${formattedSql}
-"""
-
-${entityName}_df = pandas.read_gbq(
-    dataset_${entityName}_sql,
-    dialect="standard",
-    use_bqstorage_api=("BIGQUERY_STORAGE_API_ENABLED" in os.environ),
-    progress_bar_type="tqdm_notebook")
-
-${entityName}_df.head(5)
+{
+  "metadata": {
+    "kernelspec": {
+      "display_name": "Python 3",
+      "language": "python",
+      "name": "python3"
+    }
+  },
+  "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "import pandas\n",
+        "import os\n",
+        "\n",
+        "# This query was generated for dataset ${underlayName} in study ${studyName} at ${timestamp}.\n",
+        "${entityName}_sql = \"\"\"\n",
+        "${formattedSql}\n",
+        "\"\"\"\n",
+        "\n",
+        "${entityName}_df = pandas.read_gbq(\n",
+        "    dataset_${entityName}_sql,\n",
+        "    dialect=\"standard\",\n",
+        "    use_bqstorage_api=(\"BIGQUERY_STORAGE_API_ENABLED\" in os.environ),\n",
+        "    progress_bar_type=\"tqdm_notebook\")\n",
+        "\n",
+        "${entityName}_df.head(5)\n"
+      ]
+    }
+  ],
+  "nbformat": 4,
+  "nbformat_minor": 2
+}