From fb5af8ad7cb27133f10b5de9e56e546b670eab4f Mon Sep 17 00:00:00 2001
From: harshsoni2024 <64592571+harshsoni2024@users.noreply.github.com>
Date: Wed, 16 Apr 2025 11:34:26 +0530
Subject: [PATCH] bigquery lib fix (#20849)

---
 ingestion/setup.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/ingestion/setup.py b/ingestion/setup.py
index 8f471a75d48..c7304bf8051 100644
--- a/ingestion/setup.py
+++ b/ingestion/setup.py
@@ -65,6 +65,7 @@ VERSIONS = {
     "pyiceberg": "pyiceberg==0.5.1",
     "google-cloud-bigtable": "google-cloud-bigtable>=2.0.0",
     "pyathena": "pyathena~=3.0",
+    "sqlalchemy-bigquery": "sqlalchemy-bigquery>=1.2.2",
 }
 
 COMMONS = {
@@ -330,7 +331,11 @@ plugins: Dict[str, Set[str]] = {
     },
     "sagemaker": {VERSIONS["boto3"]},
     "salesforce": {"simple_salesforce~=1.11", "authlib>=1.3.1"},
-    "sample-data": {VERSIONS["avro"], VERSIONS["grpc-tools"]},
+    "sample-data": {
+        VERSIONS["avro"],
+        VERSIONS["grpc-tools"],
+        VERSIONS["sqlalchemy-bigquery"],
+    },
     "sap-hana": {"hdbcli", "sqlalchemy-hana"},
     "sas": {},
     "singlestore": {VERSIONS["pymysql"]},