From e191034c1800e53378542adcbcd6eba3a9f5303c Mon Sep 17 00:00:00 2001
From: Prajwal214 <167504578+Prajwal214@users.noreply.github.com>
Date: Thu, 9 May 2024 08:57:25 +0530
Subject: [PATCH] Minor: Updated Python Dependency for GreenPlum (#16139)

---
 ingestion/setup.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/ingestion/setup.py b/ingestion/setup.py
index 80687a27b4a9..4f1a8b0418eb 100644
--- a/ingestion/setup.py
+++ b/ingestion/setup.py
@@ -81,6 +81,12 @@
         ],  # grpcio-tools already depends on grpcio. No need to add separately
         "protobuf",
     },
+    "postgres": {
+        VERSIONS["pymysql"],
+        "psycopg2-binary",
+        VERSIONS["geoalchemy2"],
+        VERSIONS["packaging"],
+    },  # Adding as Postgres SQL & GreenPlum are using common packages.
 }
 
 
@@ -184,6 +190,7 @@
     },  # also requires requests-aws4auth which is in base
     "glue": {VERSIONS["boto3"]},
     "great-expectations": {VERSIONS["great-expectations"]},
+    "greenplum": {*COMMONS["postgres"]},
     "hive": {
         *COMMONS["hive"],
         "thrift>=0.13,<1",
@@ -226,12 +233,7 @@
     "oracle": {"cx_Oracle>=8.3.0,<9", "oracledb~=1.2"},
     "pgspider": {"psycopg2-binary", "sqlalchemy-pgspider"},
     "pinotdb": {"pinotdb~=0.3"},
-    "postgres": {
-        VERSIONS["pymysql"],
-        "psycopg2-binary",
-        VERSIONS["geoalchemy2"],
-        VERSIONS["packaging"],
-    },
+    "postgres": {*COMMONS["postgres"]},
     "powerbi": {
         VERSIONS["msal"],
         VERSIONS["boto3"],