use tqdm.auto for optimal progressbar display

bqth29 · Oct 5, 2024 · a52f85a · a52f85a
1 parent a9ab8bb
commit a52f85a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/simulated_bifurcation/optimizer/simulated_bifurcation_optimizer.py b/src/simulated_bifurcation/optimizer/simulated_bifurcation_optimizer.py
@@ -5,7 +5,7 @@
 
 import torch
 from numpy import minimum
-from tqdm import tqdm
+from tqdm.auto import tqdm
 
 from .environment import ENVIRONMENT
 from .simulated_bifurcation_engine import SimulatedBifurcationEngine

diff --git a/src/simulated_bifurcation/optimizer/stop_window.py b/src/simulated_bifurcation/optimizer/stop_window.py
@@ -1,7 +1,7 @@
 from typing import Tuple, Union
 
 import torch
-from tqdm import tqdm
+from tqdm.auto import tqdm
 
 
 class StopWindow: