Add tqdm

fangpenlin · fangpenlin · commit af379d8d49aa · 2025-06-14T16:33:28.000-07:00
diff --git a/tests/test_loader.py b/tests/test_loader.py
@@ -74,7 +74,7 @@ def test_share_memory_shim():
             memory_pool_block_count=num_worker,
         )
         with load_with_workers(loader, range(n), num_worker) as generator:
-            for x, y in generator:
+            for x, y in tqdm.tqdm(generator):
                 assert x.numpy().shape == data_size
                 assert y.numpy().shape == label_size
                 count += 1

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,7 @@ def test_share_memory_shim():`
`74`	`74`	`memory_pool_block_count=num_worker,`
`75`	`75`	`)`
`76`	`76`	`with load_with_workers(loader, range(n), num_worker) as generator:`
`77`		`- for x, y in generator:`
	`77`	`+ for x, y in tqdm.tqdm(generator):`
`78`	`78`	`assert x.numpy().shape == data_size`
`79`	`79`	`assert y.numpy().shape == label_size`
`80`	`80`	`count += 1`