fix optional tqdm

xoolive · Sep 26, 2024 · ae248ed · ae248ed
1 parent 014b685
commit ae248ed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/traffic/data/datasets/default.py b/src/traffic/data/datasets/default.py
@@ -3,9 +3,9 @@
 from typing import TypedDict
 
 import httpx
-from tqdm.rich import tqdm
 
 from ... import cache_dir
+from ...core import tqdm
 
 client = httpx.Client(follow_redirects=True)
 
@@ -35,7 +35,7 @@ def get_data(self, entry: Entry) -> Path:
                         file_handle.write(content)
                         md5_hash.update(content)
                     else:
-                        with tqdm(
+                        with tqdm(  # type: ignore
                             total=int(content_length),
                             unit_scale=True,
                             unit_divisor=1024,