def benchmark_status(database, interval): init_db(database) total = get_total_count() last = get_done_count() progress = tqdm(total=total, initial=last) while last < total: time.sleep(interval) current = get_done_count() progress.update(current - last) last = current
def update(config=None, output_dir=None, repeat=1): db_path = get_db_path(output_dir) init_db(db_path) update_steps(config)
def __init__(self, output_dir, config, **kwargs): self.output_dir = output_dir self.config = read_config(config) self.db_path = get_db_path(output_dir) init_db(self.db_path)
def bootstrap_db(output_dir): db_path = get_db_path(output_dir) init_db(db_path) db.connect() db.create_tables(MODELS, safe=True)