exchanged napari progress with tqdm (#418)

constantinpape · Nov 18, 2024 · 8850d5e · 8850d5e
1 parent 4f7b180
commit 8850d5e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/torch_em/util/prediction.py b/torch_em/util/prediction.py
@@ -4,7 +4,10 @@
 import nifty.tools as nt
 import numpy as np
 import torch
-from tqdm import tqdm
+try:
+    from napari.utils import progress as tqdm
+except ImportError:
+    from tqdm import tqdm
 
 from ..transform.raw import standardize