Set iterator based upon tqdm

pybamm-team · Nov 13, 2023 · fd57f0f · fd57f0f
1 parent a5d2573
commit fd57f0f
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/pybamm/simulation.py b/pybamm/simulation.py
@@ -532,7 +532,10 @@ def solve(
             Additional key-word arguments passed to `solver.solve`.
             See :meth:`pybamm.BaseSolver.solve`.
         """
-        tqdm = have_optional_dependency("tqdm")
+        try:
+            tqdm = have_optional_dependency("tqdm")
+        except ModuleNotFoundError:
+            tqdm = False
         # Setup
         if solver is None:
             solver = self._solver
@@ -727,13 +730,18 @@ def solve(
                         # Update _solution
                         self._solution = current_solution
 
-            for cycle_num, cycle_length in enumerate(
-                # tqdm is the progress bar.
-                tqdm.tqdm(
+            if tqdm:
+                iterator = tqdm.tqdm(
                     self.experiment.cycle_lengths,
                     disable=(not showprogress),
                     desc="Cycling",
-                ),
+                )
+            else:
+                iterator = self.experiment.cycle_lengths
+
+            for cycle_num, cycle_length in enumerate(
+                # tqdm is the progress bar.
+                iterator,
                 start=1,
             ):
                 logs["cycle number"] = (