progress bar

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