feat: conditional tqdm helper

DigitalProductInnovationAndDevelopment · Jul 14, 2024 · ad76e98 · ad76e98
1 parent e2e695b
commit ad76e98
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/utils/tqdm_utils.py b/src/utils/tqdm_utils.py
@@ -0,0 +1,7 @@
+from tqdm import tqdm
+
+
+def conditional_tqdm(iterable, condition=True, **kwargs):
+    if condition:
+        return tqdm(iterable, **kwargs)
+    return iterable