From d6d98d20cb1ef7a93fba15a1c810bbafa2ad8f77 Mon Sep 17 00:00:00 2001
From: beerpsi <beerpsi@duck.com>
Date: Thu, 12 Dec 2024 20:47:34 +0700
Subject: [PATCH] fix: typing shenanigans

---
 core/data/database.py    | 2 +-
 core/data/schema/base.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/core/data/database.py b/core/data/database.py
index bed3e79..fb36ebd 100644
--- a/core/data/database.py
+++ b/core/data/database.py
@@ -25,7 +25,7 @@ from core.utils import MISSING, Utils
 
 class Data:
     engine: ClassVar[AsyncEngine] = MISSING
-    session: ClassVar[sessionmaker[AsyncSession]] = MISSING
+    session: ClassVar["sessionmaker[AsyncSession]"] = MISSING
     user: ClassVar[UserData] = MISSING
     arcade: ClassVar[ArcadeData] = MISSING
     card: ClassVar[CardData] = MISSING
diff --git a/core/data/schema/base.py b/core/data/schema/base.py
index adacb9c..80ab74b 100644
--- a/core/data/schema/base.py
+++ b/core/data/schema/base.py
@@ -39,7 +39,7 @@ event_log: Table = Table(
 
 
 class BaseData:
-    def __init__(self, cfg: CoreConfig, conn: sessionmaker[AsyncSession]) -> None:
+    def __init__(self, cfg: CoreConfig, conn: "sessionmaker[AsyncSession]") -> None:
         self.config = cfg
         self.conn = conn
         self.logger = logging.getLogger("database")