Skip to content

Commit 0a9dcb8

Browse files
committed
chore: add type hint
1 parent 7cf08e9 commit 0a9dcb8

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

db.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
@dataclass
88
class DB:
9-
connector: Union[sqlite3.Connection]
9+
connector: Union[sqlite3.Connection, psycopg.Connection]
1010
cursor: Union[sqlite3.Cursor, psycopg.Cursor]
1111
db_config: DatabaseConfig
1212

@@ -103,7 +103,7 @@ def get_latest_translations(cls, src_lang: str, tgt_lang: str, index: int):
103103

104104
cls.cursor.execute(cls._fill_placeholder(query), (src_lang, tgt_lang, index))
105105
records = cls.cursor.fetchall()
106-
106+
107107
return [TranslationRecord(record[0], record[1], record[2], record[3]) for record in records]
108108

109109
@classmethod

0 commit comments

Comments
 (0)