rich tqdm

rich_tqdm.py

from rich.progress import (
    ProgressColumn,
    BarColumn,
    DownloadColumn,
    TextColumn,
    TransferSpeedColumn,
    TimeRemainingColumn,
    Progress,
    TaskID,
    TimeElapsedColumn,
    SpinnerColumn,
    Text
)

class SpeedColumn(ProgressColumn):
    """Renders human readable transfer speed."""

    def render(self, task: "Task") -> Text:
        """Show data transfer speed."""
        speed = task.speed
        if speed is None:
            return Text("?", style="progress.data.speed")
        return Text(f"{speed:2.2f} it/s", style="progress.data.speed")

def RichTQDM():
    return Progress(
    SpinnerColumn(),
    "[progress.description]{task.description}",
    BarColumn(),
    TextColumn("{task.completed}/{task.total}"),
    "[",
    TimeElapsedColumn(),
    "<",
    TimeRemainingColumn(),
    ',',
    SpeedColumn(),
    ']',
    refresh_per_second=.1, speed_estimate_period=30
    )
  
if __name__ == "__main__":
    with RichTQDM() as prog:
        task1 = prog.add_task("[red]steps", total=num_steps)
        for i in range(num_steps):
            prog.update(task1, advance=1)