DISC: use rich instead of TQDM

[Xylambda]

MRE

import time
from rich.progress import Progress, BarColumn, TimeRemainingColumn


train_loader = [1, 2, 3, 4]
val_loader = [5, 6]

for epoch in range(0, 10):
    pr = Progress(
        "[progress.description]{task.description}",
        BarColumn(),
        "[progress.percentage]{task.percentage:>3.0f}%",
        TimeRemainingColumn(),
    )
    with pr as progress:
        task = progress.add_task(
            description=f'Epoch {epoch+1}/10',
            total=len(train_loader) + len(val_loader)
        )
        
        for t in train_loader:
            time.sleep(0.3)
            #progress.console.print(f"Working on job: {t}")
            progress.update(task, advance=1) # , description=f'Job {t}'
            
        for t in val_loader:
            time.sleep(0.3)
            progress.update(task, advance=1)