tweak tqdm import guard

DanielYang59 · Oct 20, 2024 · 0c9d844 · 0c9d844
1 parent 6f7fdbd
commit 0c9d844
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/monty/multiprocessing.py b/src/monty/multiprocessing.py
@@ -9,8 +9,8 @@
 
 try:
     from tqdm.autonotebook import tqdm
-except ImportError:
-    tqdm = None
+except ImportError as exc:
+    raise ImportError("tqdm must be installed for this function.") from exc
 
 
 def imap_tqdm(nprocs: int, func: Callable, iterable: Iterable, *args, **kwargs) -> list:
@@ -28,8 +28,6 @@ def imap_tqdm(nprocs: int, func: Callable, iterable: Iterable, *args, **kwargs)
     Returns:
         Results of Pool.imap.
     """
-    if tqdm is None:
-        raise ImportError("tqdm must be installed for this function.")
     data = []
     with Pool(nprocs) as pool:
         try: