update to use psycopg

CityofToronto · Aug 2, 2024 · c3176cc · c3176cc
1 parent cd329d4
commit c3176cc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/rick.py b/rick.py
@@ -2,8 +2,8 @@
 """
 Version 0.8.0
 """
-from psycopg2 import connect
-import psycopg2.sql as pg
+from psycopg import connect
+import psycopg.sql as pg
 import pandas.io.sql as pandasql
 import matplotlib as mpl
 import matplotlib.pyplot as plt
@@ -1643,4 +1643,4 @@ def general_grouped_bar_chart(data:pd.DataFrame, param_axis:str, index_axis:str,
         )
 
 
-        return fig, ax
+        return fig, ax