leo/khqp_trainer
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

优化import

Browse Source
This commit is contained in:
leo
2024-03-06 20:54:15 +08:00
parent ff1d50bc3b
commit 8d4eb36893
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
db_utils.py
View File

@@ -1,12 +1,8 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# -*- coding:utf-8 -*- # -*- coding:utf-8 -*-
from loguru import logger
from sqlalchemy import Column, Integer, String, create_engine from sqlalchemy import Column, Integer, String, create_engine
from sqlalchemy.orm import declarative_base, sessionmaker from sqlalchemy.orm import declarative_base, sessionmaker
from bak.evolve_config2 import train_data
from bak.init_data import BatchDataBase
Base = declarative_base() Base = declarative_base()
# 创建 SQLite 数据库引擎 # 创建 SQLite 数据库引擎
engine = create_engine('sqlite:///data.db', echo=False) engine = create_engine('sqlite:///data.db', echo=False)
@@ -30,4 +26,4 @@ class BatchData(Base):
Base.metadata.create_all(engine) Base.metadata.create_all(engine)
Session = sessionmaker(bind=engine, ) Session = sessionmaker(bind=engine, )
session = Session() session = Session()