From b627c1052d7dd6a6a73214777a7c4893b9e5afa1 Mon Sep 17 00:00:00 2001
From: csunny <cfqsunny@163.com>
Date: Mon, 25 Mar 2024 20:45:41 +0800
Subject: [PATCH] fix: pg py dependency problem

---
 setup.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index b982cb680..a0efbd927 100644
--- a/setup.py
+++ b/setup.py
@@ -561,7 +561,6 @@ def all_datasource_requires():
     setup_spec.extras["datasource"] = [
         # "sqlparse==0.4.4",
         "pymysql",
-        "psycopg2",
         # for doris
         # mysqlclient 2.2.x have pkg-config issue on 3.10+
         "mysqlclient==2.1.0",
@@ -569,6 +568,9 @@ def all_datasource_requires():
     setup_spec.extras["datasource_all"] = setup_spec.extras["datasource"] + [
         "pyspark",
         "pymssql",
+        # install psycopg2-binary when you are in a virtual environment
+        # pip install psycopg2-binary
+        "psycopg2",
         # pydoris is too old, we should find a new package to replace it.
         "pydoris>=1.0.2,<2.0.0",
         "clickhouse-connect",