3D view of simulator state

[cisimon7]

Is there a way to change the view of the simulator state and have something like in the waymax paper with the dark theme and colors:

At the moment, what I am getting seem to be the default settings like below:

I noticed I can pass a viz_config when calling visualization.plot_simulator_state(...) function, but the config seem not to have any key word for this.

[JinQiangWang2021]

@cisimon7
Hi have you solve this question?