Adisri99 commited on
Commit
0f8837f
·
verified ·
1 Parent(s): 1ef7d85

Update scripts/migrate_db.py

Browse files
Files changed (1) hide show
  1. scripts/migrate_db.py +4 -5
scripts/migrate_db.py CHANGED
@@ -1,6 +1,5 @@
1
- from backend.app.db import Base, engine
2
- from backend.app.models.dataset import Dataset
3
- from backend.app.models.experiment import ClusterResult, Experiment
4
 
5
- Base.metadata.create_all(bind=engine)
6
- print("Database initialized.")
 
 
1
+ from backend.app.db import init_db
 
 
2
 
3
+ if __name__ == "__main__":
4
+ init_db()
5
+ print("Database tables created.")