From 3d35affec92c4936d68c41c235a3d0de29a9775b Mon Sep 17 00:00:00 2001
From: Phionx <shanjha@mit.edu>
Date: Tue, 4 Jun 2024 00:15:34 -0400
Subject: [PATCH] updated bar format

---
 jaxquantum/core/solvers.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/jaxquantum/core/solvers.py b/jaxquantum/core/solvers.py
index 573eb3e..2ad14c0 100644
--- a/jaxquantum/core/solvers.py
+++ b/jaxquantum/core/solvers.py
@@ -67,7 +67,8 @@ def spre(op: Array) -> Callable[[Array], Array]:
 class CustomProgressMeter(TqdmProgressMeter):
     @staticmethod
     def _init_bar() -> tqdm.tqdm:
-        return tqdm.tqdm(total=100, unit='%', colour="MAGENTA", ascii="░▒█")
+        bar_format = "{desc}: {percentage:3.0f}% |{bar}| [{elapsed}<{remaining}, {rate_fmt}{postfix}]"
+        return tqdm.tqdm(total=100, bar_format=bar_format, unit='%', colour="MAGENTA", ascii="░▒█")
     
 
 def solve(ρ0, f, t_list, args, solver_options):