Rich progress bars instead of tqdm

Signed-off-by: Devansh Agarwal <[email protected]>
thepetabyteproject · Oct 15, 2020 · f7515f7 · f7515f7
1 parent b6a4111
commit f7515f7
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/your/writer.py b/your/writer.py
@@ -188,7 +188,9 @@ def to_fil(self):
         self.outname += ".fil"
         with Progress() as progress:
             if not self.progress:
-                task = progress.add_task("[green]Writing...", total=self.nsamp, visible=False)
+                task = progress.add_task(
+                    "[green]Writing...", total=self.nsamp, visible=False
+                )
             else:
                 task = progress.add_task("[green]Writing...", total=self.nsamp)
             # create the header
@@ -269,7 +271,9 @@ def to_fits(self, npsub=-1):
         st = self.nstart
         with Progress() as progress:
             if not self.progress:
-                task = progress.add_task("[green]Writing...", total=nsubints, visible=False)
+                task = progress.add_task(
+                    "[green]Writing...", total=nsubints, visible=False
+                )
             else:
                 task = progress.add_task("[green]Writing...", total=nsubints)
 
@@ -281,7 +285,9 @@ def to_fits(self, npsub=-1):
                     pass
                 isub = istop - istart
 
-                logger.info(f"Writing data to {outfile} from subint = {istart} to {istop}.")
+                logger.info(
+                    f"Writing data to {outfile} from subint = {istart} to {istop}."
+                )
 
                 # Read in nread samples from filfile
                 nread = isub * npsub
@@ -385,7 +391,9 @@ def to_dada(self):
         header = self.dada_header()
         with Progress() as progress:
             if not self.progress:
-                task = progress.add_task("[green]Reading...", total=self.nsamp, visible=False)
+                task = progress.add_task(
+                    "[green]Reading...", total=self.nsamp, visible=False
+                )
             else:
                 task = progress.add_task("[green]Reading...", total=self.nsamp)