From 6bb4a64624a5c7c17675c899003826cd3172dc66 Mon Sep 17 00:00:00 2001 From: mm Date: Fri, 5 May 2023 06:21:28 +0000 Subject: [PATCH] progress bar --- generate_data.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/generate_data.py b/generate_data.py index f580a6e..969f091 100644 --- a/generate_data.py +++ b/generate_data.py @@ -8,6 +8,7 @@ from functools import lru_cache import geonamescache import numpy as np from geopy.distance import geodesic +from tqdm import tqdm MAX_DISTANCE = 20_037.5 @@ -126,8 +127,13 @@ def main(): try: executor = concurrent.futures.ProcessPoolExecutor(max_workers=args.workers) - for i in range(num_chunks): - print(f"Processing chunk {i}...") + for i in tqdm( + range(num_chunks), + total=num_chunks, + desc="Processing chunks", + ncols=100, + bar_format="{l_bar}{bar:30}{r_bar}", + ): chunk = city_combinations[(i * chunk_size) : (i + 1) * chunk_size] futures = { executor.submit(calculate_distance, pair): pair for pair in chunk