A request for visualization of Antmaze

Hi, good job! Your open source has helped me a lot!

I would like to ask if I need to visualize a trajectory like Figure 18 of HIQL paper. How can i do?  And I just want to visualize my method. Could you give me some advice?

Thanks!