fix: benchmark_migration.py needs to close sssion (#15822)

This commit is contained in:
Beto Dealmeida
2021-07-21 11:16:46 -07:00
committed by GitHub
parent 32a5680510
commit d26254099e
2 changed files with 5 additions and 2 deletions

View File

@@ -172,6 +172,7 @@ def main(
rows = session.query(model).count()
print(f"- {model.__name__} ({rows} rows in table {model.__tablename__})")
model_rows[model] = rows
session.close()
print("Benchmarking migration")
results: Dict[str, float] = {}