diff --git a/db_utils.py b/db_utils.py index 8ce0a17..4c2f1bb 100644 --- a/db_utils.py +++ b/db_utils.py @@ -1,12 +1,8 @@ #!/usr/bin/env python3 # -*- coding:utf-8 -*- -from loguru import logger from sqlalchemy import Column, Integer, String, create_engine from sqlalchemy.orm import declarative_base, sessionmaker -from bak.evolve_config2 import train_data -from bak.init_data import BatchDataBase - Base = declarative_base() # 创建 SQLite 数据库引擎 engine = create_engine('sqlite:///data.db', echo=False) @@ -30,4 +26,4 @@ class BatchData(Base): Base.metadata.create_all(engine) Session = sessionmaker(bind=engine, ) -session = Session() \ No newline at end of file +session = Session()