error with customizing my progress par

[abdelkareemkobo]

I am trying to use RichProgressBar !
when i run the following code , I get this error

from pytorch_lightning.callbacks import RichProgressBar
from rich.progress import Progress
model = ButterflyGAN()
trainer = pl.Trainer( max_epochs=100, gpus=1,callbacks=[RichProgressBar(refresh_rate=1,
                                            theme=RichProgressBarTheme(description='black'
                                                                        ,progressbar = '#6206E0'
                                                                        ,progress_bar_finished='#6206E0'
                                                                        ,progress_bar_pulse='#6206E0',
                                                                        batch_size='white',
                                                                        time='grey54',
                                                                        processing_speed='grey70',
                                                                        metrics='white'),console_kwargs=None)])#gpus=1
trainer.fit(model, buttefly_train_dataloader)

NameError: name'RichProgressBarTheme' is not defined 

any help

[akihironitta]

You need to import RichProgressBar before instantiating the class.

from pytorch_lightning.callbacks.progress.rich_progress import RichProgressBarTheme

https://pytorch-lightning.readthedocs.io/en/1.7.7/visualize/logging_expert.html?highlight=RichProgressBarTheme#use-the-richprogressbar