@@ -1455,7 +1455,8 @@ class PandasSQL(PandasObject, ABC):
1455
1455
def __enter__ (self ) -> Self :
1456
1456
return self
1457
1457
1458
- def __exit__ (self , * args ) -> None :
1458
+ @abstractmethod
1459
+ def __exit__ (self , exc_type , exc_value , traceback ) -> None :
1459
1460
pass
1460
1461
1461
1462
def read_table (
@@ -1647,7 +1648,7 @@ def __init__(
1647
1648
self .meta = MetaData (schema = schema )
1648
1649
self .returns_generator = False
1649
1650
1650
- def __exit__ (self , * args ) -> None :
1651
+ def __exit__ (self , exc_type , exc_value , traceback ) -> None :
1651
1652
if not self .returns_generator :
1652
1653
self .exit_stack .close ()
1653
1654
@@ -2126,6 +2127,9 @@ class ADBCDatabase(PandasSQL):
2126
2127
def __init__ (self , con ) -> None :
2127
2128
self .con = con
2128
2129
2130
+ def __exit__ (self , exc_type , exc_value , traceback ) -> None :
2131
+ self .con .close ()
2132
+
2129
2133
@contextmanager
2130
2134
def run_transaction (self ):
2131
2135
with self .con .cursor () as cur :
@@ -2679,6 +2683,9 @@ class SQLiteDatabase(PandasSQL):
2679
2683
def __init__ (self , con ) -> None :
2680
2684
self .con = con
2681
2685
2686
+ def __exit__ (self , exc_type , exc_value , traceback ) -> None :
2687
+ self .con .close ()
2688
+
2682
2689
@contextmanager
2683
2690
def run_transaction (self ):
2684
2691
cur = self .con .cursor ()
0 commit comments