in evalbench/databases/db.py [0:0]
def resetup_database(self, force=False, setup_users=False) -> None:
if not self.setup_scripts:
raise ValueError("Setup sql is required for setup.")
if self.was_re_setup_this_session and not force:
# If database was already re-setup (for DQL, DML, etc.)
# and it can be re-used, don't re-run setup unless forced.
return
pre_setup, setup, post_setup = self.setup_scripts
self.drop_all_tables()
self.batch_execute(pre_setup)
self.batch_execute(setup)
self.insert_data(self.data, setup)
self.batch_execute(post_setup)
if setup_users:
self.setup_tmp_users()
self.was_re_setup_this_session = True