Skip to content

Commit 28decbd

Browse files
committed
Pass test locally.
1 parent febfc79 commit 28decbd

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

clients/database.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from pandas.core.groupby.generic import DataFrameGroupBy
66
from sqlalchemy import create_engine
77
from sqlalchemy.types import DateTime, Integer, String, Text
8+
from sqlalchemy.engine import Engine
89

910

1011
class Database:
@@ -16,8 +17,12 @@ def __init__(self, uri, table, args=None):
1617
self.args = args
1718

1819
@property
19-
def engine(self):
20-
"""Database connection engine"""
20+
def engine(self) -> Engine:
21+
"""
22+
Database connection engine
23+
24+
:returns: Engine
25+
"""
2126
return create_engine(self.uri, connect_args=self.args)
2227

2328
def upload_dataframe(self, df: DataFrame) -> str:

0 commit comments

Comments
 (0)