docs: use os.getenv instead of google.colab

pingcap · Mar 3, 2025 · 98f992e · 98f992e
1 parent d337daf
commit 98f992e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/core/tests/quickstart.ipynb b/core/tests/quickstart.ipynb
@@ -86,10 +86,10 @@
    "source": [
     "from sqlalchemy import create_engine\n",
     "from autoflow import Autoflow\n",
-    "from google.colab import userdata\n",
+    "# from google.colab import userdata\n",
+    "# db_engine = create_engine(userdata.get('DATABASE_URL'))\n",
     "\n",
-    "# db_engine = create_engine(os.getenv(\"DATABASE_URL\"))\n",
-    "db_engine = create_engine(userdata.get('DATABASE_URL'))\n",
+    "db_engine = create_engine(os.getenv(\"DATABASE_URL\"))\n",
     "af = Autoflow(db_engine=db_engine)"
    ]
   },